python 用 read_csv讀取資料集時刪除某幾列元素
阿新 • • 發佈:2021-02-01
首先用把所有的資料讀進來並轉換成列表的形式。
讀進來的資料格式為列表套列表。
比如說我們不想要資料列表中每個元素的第一和第二個(以鶯尾花資料為例)。我們可以這樣做:
def ignore_column(complete_data, column_list):
"""
因為每刪除一個,列表就會變短一個,所以除了column_list第一個,我們都讓column_list的元素減一
:param complete_data: 原始資料集
:param column_list:要刪除的列的列表
:return:
"""
for i in range(len(complete_data)):
is_not_first = False
for j in column_list:
if is_not_first:
j = int(j) - 1
del complete_data[i][j]
is_not_first = True
return complete_data
刪除第一列和第二列的結果如下: