使用pandas从python中现有的csv文件生成指定数量的新csv文件数据帧的方法

Question

我在Csv文件样本1中有一个大数据帧，因此我必须生成一个仅包含100个数据帧的新Csv文件。我为它生成了代码。但是我得到了关键错误标签[100]不在索引中？ 我刚刚试过如下，任何帮助都将不胜感激

import pandas as pd
data_frame = pd.read_csv("C:/users/raju/sample1.csv")
data_frame1 = data_frame[:100]
data_frame.to_csv("C:/users/raju/sample.csv")`

`

Answer 1

正确的语法是^{}：

data_frame.iloc[:100]

一种更有效的方法是使用nrows参数，其目的正是提取文件的一部分。这样可以避免在分析无用行时浪费资源和时间：

import pandas as pd
data_frame = pd.read_csv("C:/users/raju/sample1.csv", nrows=101)  # 100+1 for header
data_frame.to_csv("C:/users/raju/sample.csv")