pycharm对csv的操作记录（不定期更新）

目录

1、查询csv的行数：

2、对某一特征为特定值的数据行

3、对获得的数据进行存储

4、csv读取和取唯一值

---

1、查询csv的行数：

print( stock_data.loc[stock_data['Name'] == 'A'].shape[0])
print( stock_data.shape[0])

2、对某一特征为特定值的数据行

stock_data_company = stock_data.loc[stock_data['Name'] == company_name]

3、对获得的数据进行存储

index=False是不保存序列号：0,1,2等。filepath：新创建的文件名

filepath = 'company_stockdata/' + company_name + '.csv'
stock_data_company.to_csv(filepath, index=False, encoding='gbk')

4、csv读取和取唯一值

stock_data = pd.read_csv('stockData.csv')
company_names = stock_data['Name'].unique()

5、将某列变成数组

top_Name=top_five['Name'].values
或
top_totalvolume=top_five['totalvolume']
top_totalvolume=np.array(top_totalvolume)

但是有个不一样的是这样子输出的数组是空格分隔的，在做表是并识别不了

可以使用.tolist(),但是对于结果丢失了2D结构。 但我这里使用的是一维，所以没有影响

top_Name=top_five['Name'].values.tolist()
top_totalvolume=top_five['totalvolume']
top_totalvolume=np.array(top_totalvolume).tolist()

注意事项

错误代码：

（1）错误：print(stock_data[3])

正确：print(stock_data[1：3])