對不起，但我不能使用熊貓。
我有一個示例輸入 csv 檔案，如下所示：
Alfa,Beta,Charlie,Delta,Echo,Foxtrot,Golf,Hotel,India,Juliett,Kilo
A1,B1,C1,D1,E1,F1,G1,H1,I1,J1,K1
A2,B2,C2,D2,E2,F2,G2,H2,I2,J2,K2
A3,B3,C3,D3,E3,F3,G3,H3,I3,J3,K3
A4,B4,C4,D4,E4,F4,G4,H4,I4,J4,K4
A5,B5,C5,D5,E1,F5,G5,H5,I5,J5,K5
A6,B6,C6,D6,E6,F6,G6,H6,I6,J6,K6
A7,B7,C7,D7,E7,F7,G7,H7,I7,J7,K7
A8,B8,C8,D8,E8,F8,G8,H8,I8,J8,K8
A9,B9,C9,D9,E9,F9,G9,H9,I9,J9,K9
然后我的示例代碼如下所示：
import csv
fieldnames_dict = {
'Beta': 'Beta_New',
'Echo': 'Echo_New',
'Foxtrot': 'Foxtrot_New_ALL',
'Hotel': 'Hotel_New',
'India': 'India_New',
'Charlie': 'Charlie_New'
}
with open("book1.csv", "r", encoding="utf-8", errors='ignore') as csv_in:
with open("xtest_file.csv", "w", encoding="utf-8", errors='ignore') as csv_out:
reader = csv.DictReader(csv_in, delimiter=',', quotechar='"')
writer = csv.DictWriter(csv_out, delimiter=',', quotechar='"',
fieldnames=list(fieldnames_dict.values()))
writer.writeheader()
for row_in in reader:
row_out = {new: row_in[old] for old, new in fieldnames_dict.items()}
writer.writerow(row_out)
所以這就是輸出 csv 應該的樣子：
Beta_New,Echo_New,Foxtrot_New_ALL,Tango,Victor,Hotel_New,India_New,Charlie_New
B1,E1,F1,T1,V1,H1,I1,C1
B2,E2,F2,T1,V1,H2,I2,C2
B3,E3,F3,T1,V1,H3,I3,C3
B4,E4,F4,T1,V1,H4,I4,C4
B5,E5,F5,T1,V1,H5,I5,C5
B6,E6,F6,T1,V1,H6,I6,C6
B7,E7,F7,T1,V1,H7,I7,C7
B8,E8,F8,T1,V1,H8,I8,C8
B9,E9,F9,T1,V1,H9,I9,C9
所以字典中的所有鍵都必須是唯一的（因此在 in 中添加兩個空字串鍵fieldnames_dict是不可能的）。也就是說，fieldnames_dict由于字典理解中的 row_in[old] ，所有鍵 in 必須已經存在于輸入 CSV 檔案中{new: row_in[old] for old, new in fieldnames_dict.items()}。這意味著我必須在最后附加/添加新鍵，但是如何在“Foxtrot_New_ALL”和“Hotel_New”之間添加鍵“Tango，Victor”和值？
uj5u.com熱心網友回復：
我覺得你誤解了它的DictWriter作業原理。
當您指定fieldnames=...時，DictWriter并不關心這些欄位名稱來自何處，只要它是字串串列即可。
當您將字典傳遞給 時writerow()，會DictWriter查看fieldnames您之前給它的 ，以決定從該字典中選擇哪些鍵以及以什么順序選擇。
所以重要的是
fieldnames是按您希望輸出檔案中的列的順序排列的列名串列。- 您傳遞給
writerow的任何 dict 都包含fieldnames. 列順序由 指定fieldnames，因此此 dict 的鍵的順序無關緊要。
考慮到這一點，您可以創建一個包含新列名稱的串列，該串列與告訴您如何重命名舊列的字典分開。
fields = ['Beta_New',
'Echo_New',
'Foxtrot_New_ALL',
'Tango',
'Victor',
'Hotel_New',
'India_New',
'Charlie_New']
column_rename_dict = {
'Beta': 'Beta_New',
'Echo': 'Echo_New',
'Foxtrot': 'Foxtrot_New_ALL',
'Hotel': 'Hotel_New',
'India': 'India_New',
'Charlie': 'Charlie_New'
}
然后，您將fields其用作撰寫器的fieldnames引數，并確保您傳遞給它的任何 dicts 都包含所有這些欄位名稱。
ctr = 1
with open(...) as csv_in, open(...) as csv_out:
reader = csv.DictReader(csv_in)
writer = csv.DictWriter(csv_out, fieldnames=fields)
writer.writeheader()
for row_in in reader:
# Convert the old field names to new field names
row_out = {new: row_in[old] for old, new in column_rename_dict.items()}
# Set values for extra keys
row_out["Tango"] = f"T{ctr}"
row_out["Victor"] = f"V{ctr}"
ctr = 1
# Write the row
writer.writerow(row_out)
這為您提供了新的 csv：
Beta_New,Echo_New,Foxtrot_New_ALL,Tango,Victor,Hotel_New,India_New,Charlie_New
B1,E1,F1,T1,V1,H1,I1,C1
B2,E2,F2,T2,V2,H2,I2,C2
B3,E3,F3,T3,V3,H3,I3,C3
B4,E4,F4,T4,V4,H4,I4,C4
B5,E1,F5,T5,V5,H5,I5,C5
B6,E6,F6,T6,V6,H6,I6,C6
B7,E7,F7,T7,V7,H7,I7,C7
B8,E8,F8,T8,V8,H8,I8,C8
B9,E9,F9,T9,V9,H9,I9,C9
轉載請註明出處，本文鏈接：https://www.uj5u.com/qiye/471450.html