（這是我在 SO 上的第一篇文章，所以如果這不是正確的格式，請告訴我！）
我正在制作一個表格，當給定一個 CSV 檔案的 ID 號時，它會提取名稱，并且完全不知道從哪里開始。
示例資料：
| 身份證號碼 | 名稱 |
|---|---|
| 1 | 約翰·多伊 |
| 2 | 簡·施莫 |
| 3 | 凱恩·弗 |
所以如果我輸入 ID 號“1”，它會輸出“Jane Doe”
謝謝
uj5u.com熱心網友回復：
使用DictReaderfromcsv模塊并創建資料字典：
import csv
with open('data.csv') as csvfile:
reader = csv.DictReader(csvfile)
data = {d['ID Number']: d['Name'] for d in list(reader)}
num = input('Input an ID Number: ')
print(data.get(num, 'Not found'))
輸出：
Input an ID Number: 1
John Doe
uj5u.com熱心網友回復：
我會根據子集來做到這一點。首先將資料作為 Pandas 資料框讀入：
import pandas as pd
df = pd.read_csv("<filename>.csv")
從這里您想通過 id 查詢您的 name 列，因此（作為示例）如果您想找到與 ID = 1 關聯的名稱：
result = df[df["ID Number"] == 1]]
uj5u.com熱心網友回復：
您可以將 csv 加載到字典中，然后反復詢問用戶輸入：
file = open("example.csv", "r")
d = {}
for line in file:
split = line.strip().split(",")
d[split[0]] = split[1]
while True:
id = input("Please input an ID: ")
print(d[id])
輸出：
Please input an ID: 1
Jane Doe
Please input an ID: 3
Kane Foe
Please input an ID: 2
Jane Shmoe
uj5u.com熱心網友回復：
以下是有關如何使用熊貓進行操作的提示：
import pandas as pd
df = pd.read_csv('sample.csv')
user=df[df['ID Number']==2].iloc[0]
print(user.to_dict())
在這個例子中，我們正在尋找 id 為 2 的用戶。然后我們選擇找到的第一個專案。下一步，我們將其轉換為字典并列印出來。
您可以使用不同的功能在命令列界面中讀取，例如輸入（https://www.geeksforgeeks.org/taking-input-in-python/）。
轉載請註明出處，本文鏈接：https://www.uj5u.com/yidong/387371.html
上一篇：如何在一行上輸出帶有fortran（用逗號分隔）的CSV檔案？[復制]
下一篇：未定義主鍵，即使它已經是