將結果性別化后保存回原始的CSV文件

Question

我有一個csv，其中包含要放入python中的sexizeize軟件包的名稱列表。 我已經成功完成了此操作，但現在想將結果保存回原始的csv中。 我知道sex-api.com會這樣做，但我不想使用它（金錢並應該使用python）。

這是我的代碼：

from genderize import Genderize
import pandas as pd

df = pd.read_csv('Names.csv')
names = df['First Name']

print(Genderize().get(names))

Results: 
[{'name': 'Theresa', 'gender': 'female', 'probability': 1.0, 'count': 834}, {'name': 'Andy', 'gender': 'male', 'probability': 0.95, 'count': 3139}, {'name': 'Boris', 'gender': 'male', 'probability': 1.0, 'count': 288}, {'name': 'Rachel', 'gender': 'female', 'probability': 1.0, 'count': 4021}]

我希望我的csv文件中包含其他列：性別，概率和計數，如結果所示。 編碼新手，很抱歉，如果這是基本知識。 謝謝。

Answer 1

您可以將此結果轉換為數據幀，然后將其寫入csv：

result = [{'name': 'Theresa', 'gender': 'female', 'probability': 1.0, 'count': 834}, {'name': 'Andy', 'gender': 'male', 'probability': 0.95, 'count': 3139}, {'name': 'Boris', 'gender': 'male', 'probability': 1.0, 'count': 288}, {'name': 'Rachel', 'gender': 'female', 'probability': 1.0, 'count': 4021}]
resultDF = pd.DataFrame(result)
resultDF.to_csv('NewFile.csv') 

或者，您可以將結果直接寫入csv：

import csv
keys = result[0].keys()
with open('newFile2.csv', 'w') as output_file:
    dict_writer = csv.DictWriter(output_file, keys)
    dict_writer.writeheader()
    dict_writer.writerows(result)