使用pandas從python中的現有csv文件生成新csv文件的指定數字數據幀的方法

Question

我在Csv文件sample1中有大數據幀，我必須生成一個新的Csv文件只包含100個data-frame.i已為它生成代碼。但是我得到鍵錯誤標簽[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

正確的語法是iloc ：

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")