如何将CSV文件转换为python字典

Question

如果我有一个CSV（info.csv）文件如下：

180101，S1，-1

180101，S2,15

180102，S4，-5

180103，S1，-5

...

列分别表示“日期”，“代码”，“信息1”。

我想转换原始数据中匹配的新代码（csv文件，newcode.scv）。

如下面的新代码映射：

180101，N1

180102，N2

180103，W1

...

将它视为一对一的关系，没有重复。

所以来自它的新数据：

180101，N1，-1

180101，N2,15

180102，N2，-5

180103，W1，-5

是否有九种奇特的转换方式？

Answer 1

一种方法是使用pandas ，它专门用于矢量化计算。 纯csv方法可能基于循环并且效率低下。

import pandas as pd

df = pd.read_csv('info_in.csv', sep=', ', engine='python',
                 header=None, names=['date', 'code', 'info1'])

#      date code  info1
# 0  180101   S1     -1
# 1  180101   S2     15
# 2  180102   S4     -5
# 3  180103   S1     -5

d = {'S1': 'N1', 'S2': 'N2', 'S4': 'W1'}
df['code'] = df['code'].map(d)
df.to_csv('info_out.csv', index=False)

#      date code  info1
# 0  180101   N1     -1
# 1  180101   N2     15
# 2  180102   W1     -5
# 3  180103   N1     -5

Answer 2

标准库的csv模块有许多用于处理csv文件的有用函数，包括读取和写入它们的简单方法。 DictReader是您想要将条目转换为字典的方法。

csv模块文档