1. 程式人生 > 其它 >python 用 read_csv讀取資料集時刪除某幾列元素

python 用 read_csv讀取資料集時刪除某幾列元素

技術標籤:pythonpython

首先用把所有的資料讀進來並轉換成列表的形式。
在這裡插入圖片描述
讀進來的資料格式為列表套列表。
在這裡插入圖片描述

比如說我們不想要資料列表中每個元素的第一和第二個(以鶯尾花資料為例)。我們可以這樣做:

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

刪除第一列和第二列的結果如下:在這裡插入圖片描述