使用iloc將CSV文件拆分為多個文件

Question

我有一個CSV文件，其中包含98列（現在），其中包含其他文件的列我只想將其他文件的列拆分為新的CSV文件，就像這樣...

我在這里上傳了一個名為GenerateAndSplitSeedFile（Parts_skills_predict）.csv的csv文件，如果你願意，可以查看它。

data =pd.read_csv('MergedSeedData(Parts_skills_Durations_fix).csv')
print(len(data.columns))
data1 = data.drop(data.iloc[:, 64:], axis=1)
print(data1)
data1.to_csv('data1file.csv', index=False)

在這里，我正在閱讀CSV文件，我從98列中獲得了僅64列，並將其寫入CSV文件。

對於其他拆分的同樣操作，這里我需要從（0 - 8）和從（64 - 83）獲取列，因為我在這里完成的是：

data22 = data.drop(data.iloc[:, 8:64], axis=1)
data2 = data22.drop(data22.iloc[:, 19:], axis=1)
# print(data2)
data2.to_csv('datafile.csv', index=False)

在這里，我在第一次拆分（data22）中拆分了0-8和64-98的列，然后從data2中刪除了額外的列

問題是，如何在單行中對data2進行拆分我想從主文件中拆分0-8,64-83列，我需要在一行中寫入總共98列...

Answer 1

有多種方法可以做到：

1.） np.concatenate和np.arange

#selecting 0-8,64-83
data.iloc[:, np.concatenate((np.arange(0,8),np.arange(64,83)),axis=0))]
#This is generate a list of column indexes you want to choose

2） np.r_

data.iloc[:, np.r_[0:8,64:83]]