自学Python数据处理(四)——数据存取_CSV文件

数据存取——CSV格式

数据的存取利用了第三方库——pandas，在程序的开始要先导入该库

import pandas as pd

1. 数据存储

#生成字典型数据
data = {'points':[30,25,25],'rebounds':[10,10,5],'assits':[5,10,5]}
#将字典型数据转换成DataFrame类型
df = pd.DataFrame(data)
#存储成CSV文件,每行数据前面自动添加索引
df.to_csv('data.csv')
#存储成csv文件,去掉每行数据前面的索引
df.to_csv('data2.csv',index=None)
#追加模式mode='a', 否则,后面程序写入的数据会覆盖前面的数据
df['points'] = df['points']+1
df.to_csv('data.csv',index=None,mode='a',header=False)

2. csv文件数据读取

import pandas as pd
df1 = pd.read_csv('data.csv')
print(df1)

程序输出：

读取时指定索引列：

df2 = pd.read_csv('data.csv',index_col='rebounds')
print(df2)

程序输出：

读取时添加列名：

#添加列名
df3 = pd.read_csv('data.csv',names=['得分','篮板','助攻'])
print(df3)

程序输出：

p.s.需要注意的是，读取数据时如果重新添加“列名”的话，数据中原有的“列名”会变成一条“数据”，这一点，从“索引”的变化也能看出来。