我正在嘗試從 CSV 文件中獲取第一列（名稱）和第四列（處于活動狀態）并執行以下操作：為公司名稱創建一個條目如果 'is active' = yes 則增加值并輸出最終結果。如果“活躍”=否，則增加該數字并給我一個“活躍”、“不活躍”列表，最后有一個值。Data1 和 Data2 字段是我目前不關心的其他列。csv = Name,Data1,Data2, Is Active:Company 1,Data1,Data2,YesCompany 1,Data1,Data2,YesCompany 1,Data1,Data2,YesCompany 2,Data1,Data2,YesCompany 2,Data1,Data2,NoCompany 2,Data1,Data2,YesCompany 2,Data1,Data2,YesCompany 3,Data1,Data2,NoCompany 3,Data1,Data2,No理想的結果將采用以下格式：Company name, Yes-count, no-count我已經開始使用 csvreader 來讀取列，我可以將它們放入列表中，但我不確定如何比較和合并名稱和計數。任何幫助將不勝感激。
1 回答
大話西游666
TA貢獻1817條經驗 獲得超14個贊
一種方法，使用這個：
with open("your_csv_file", "r") as file:
reader = csv.reader(file)
_ = next(reader) # skip header
consolidated = {}
for line in reader:
company_name = line[0]
is_active = line[3]
if company_name not in consolidated:
consolidated[company_name] = { "yes_count": 0, "no_count": 0}
if is_active == "Yes":
consolidated[company_name]["yes_count"] += 1
else:
consolidated[company_name]["no_count"] += 1
樣本輸出：
>>> print(consolidated)
{
'Company 1': {'yes_count': 3, 'no_count': 0},
'Company 2': {'yes_count': 3, 'no_count': 1},
'Company 3': {'yes_count': 0, 'no_count': 2}
}
添加回答
舉報
0/150
提交
取消