pd.DataFrame
要保存为CSV文件的数据需要是一个DataFrame或Series。采用pd.DataFrame创建DataFrame
如果是字典则可直接使用
data = {
'列1': [1, 2, 3],
'列2': ['A', 'B', 'C']
}
df = pd.DataFrame(data)
如果是列表,以及Numpy数组
data = [
[1, 'A'],
[2, 'B'],
[3, 'C']
]
df = pd.DataFrame(data, columns=['列1', '列2'])
同时可指定索引
df = pd.DataFrame(data, index=['行1', '行2', '行3'])
最后可以直接转换成csv文件,index = False 的意思是保存csv文件时不将DataFrame的行号,保存到csv文件当中,就是下侧的行号
df.to_csv('filename.csv', index=False)