欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  后端开发

Pandas读写CSV文件的方法介绍(附代码)

程序员文章站 2022-05-06 11:53:17
...

本篇文章给大家带来的内容是关于Pandas读写CSV文件的方法介绍(附代码),有一定的参考价值,有需要的朋友可以参考一下,希望对你有所帮助。

读csv:使用pandas读取

import pandas as pd
import csv
if name == '__main__':
# header=0——表示csv文件的第一行默认为dataframe数据的行名称,
# index_col=0——表示使用第0列作为dataframe的行索引,
# squeeze=True——表示如果文件只包含一列,则返回一个序列。
file_dataframe = pd.read_csv('../datasets/data_new_2/csv_file_name.csv', header=0, index_col=0, squeeze=True)
# 结果:

Pandas读写CSV文件的方法介绍(附代码)

# 当参数index_col=False 时,自动生成行索引0到n

Pandas读写CSV文件的方法介绍(附代码)

# csv数据:

Pandas读写CSV文件的方法介绍(附代码)

data_1 = []
# 读取行索引一样的数据,保存为list
try:
    # 行索引为i的数据有多行,列为'pre_star'
    data_1.extend(file_dataframe .loc[i]['pre_star'].values.astype(float))
except AttributeError:
    # 行索引为i的数据只有单行,
    data_1.extend([file_dataframe .loc[i]['pre_star']])
# 多行结果

Pandas读写CSV文件的方法介绍(附代码)

# 行索引为i的数据只有一行时,不能对file_dataframe .loc[i]['pre_star']使用.values,否则会报错:

Pandas读写CSV文件的方法介绍(附代码)

写csv

使用csv写

stu1 = [lid, k, pre_count_data[k]]
# 打开文件,写模式为追加'a'
out = open('../results/write_file.csv', 'a', newline='')
# 设定写入模式
csv_write = csv.writer(out, dialect='excel')
# 写入具体内容
csv_write.writerow(stu1)

本篇文章到这里就已经全部结束了,更多其他精彩内容可以关注PHP中文网的python视频教程栏目!

以上就是Pandas读写CSV文件的方法介绍(附代码)的详细内容,更多请关注其它相关文章!

相关标签: pandas python