我如何制作一个函数来读取csv，将其列表转换为字典，从而在python中产生特定的输出

Question

python中的csv文件，其中包括整数列表和字符串，例如：

Sunset,61,South,2002-12-28,48,3,25,168,42,13

Sunrise,42,North,2012-02-05,84,3,39,141,13,115

我该如何制作一个函数，打开并读取csv，将其转换为字典(在这种情况下命名为dict(，以便在函数中输入这样的输入：

sun_state(dict, 61)

并得到输出：

[ '2002-12-28', '84', '4', '49', '308']

输出的格式分别如下：

日期，日期后的三个整数和最后三个整数的和。

如果我的问题有任何问题，请告诉我，我很乐意得到任何形式的帮助，谢谢。

Answer 1

您只需使用panda读取csv文件并创建一个字典(尽管我无法理解您的解释(，如下所示：

代码：

import pandas as pd
if __name__ == '__main__':
df = pd.read_csv("data/id_to_date.csv", header=None, names=["Sunset","ID","Direction","Date","col1","col2","col3","col4","col5","col6"])
sun_dict = {}
for row_id, row in df.iterrows():
values = [row["Date"], row["col1"], row["col2"], row["col3"], sum([row["col4"], row["col5"], row["col6"]])] 
sun_dict[row["ID"]] = values
print(sun_dict[61])

结果：

['2002-12-28', 48, 3, 25, 223]