pandas取csv 部分行，部分列数据

最新推荐文章于 2022-08-05 14:41:10 发布

原创最新推荐文章于 2022-08-05 14:41:10 发布 · 8.1k 阅读

14 ·

CC 4.0 BY-SA版权

Python 专栏收录该内容

21 篇文章

订阅专栏

本文介绍如何使用Python的pandas库进行CSV数据的读取、筛选和重新排列。具体包括选取数据集的一半行数和特定列，以及按新顺序保存到CSV文件的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import pandas as pd

csv_data = pd.read_csv('file1.csv')  # 读取数据

#取一半行的数据，并存储到新的文件
df = pd.DataFrame()
df=csv_data[0:(int)(csv_data.shape[0]/2)]
df.to_csv(path_or_buf='file2.csv',index=False)


#取某些列，按照新的顺序排列，并存储到新的文件
df = pd.DataFrame()
df=(csv_data.loc[:,['workclass', 'education',  'capital_loss', 'hours_per_week']])
df.to_csv(path_or_buf='file2.csv',index=False)