pandas使用总结

pandas获取csv，excel的标题方法可以用如下代码

df = pd.read_excel('df.xlsx')
index_name = pd.DataFrame(df).columns.values.tolist()
pandas存入csv

存入的result可以是列表，也可以是矩阵

result = [[1,2,3,],[4,5,6]]

result = pd.DataFrame(result)
result.to_csv('result.csv', index=False, header=['true_y', 'predict_y', 'probability'])
pandas随机抽取数据
df2 = pd.DataFrame(data)
df_negative = pd.DataFrame(data_negative)
df2 = df2.append(df_negative) # 把df_negative添加到df2
df2.to_csv('filter' + str(i) + '.csv', header=index_name, index=False)

随机从positive和negitave里面抽取n个样本

data_negative = pd.DataFrame(negitave).sample(negitave_size)

data = pd.DataFrame(positive).sample(int(positive_size / n))

原创文章，转载请注明出处：http://124.221.219.47/article/pandas/