csv.reader返回的數據類型詳解

一、csv.reader返回的數據類型介紹

Csv.reader()返回的是一個迭代器對象，包含了CSV文件中的每一行作為一個列表。

import csv

with open('example.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

這裡的每一行都是一個列表，列表的每個元素對應着CSV文件中的每個值。

二、讀取CSV行

我們可以使用for循環逐行讀取CSV文件中的內容。

import csv

with open('example.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

結果輸出為:

['1', 'John', 'Doe', 'example@gmail.com', '']
['2', 'Jane', 'Doe', 'jane@example.com', '']
['3', 'Bob', 'Smith', 'bob@example.com', '']

三、讀取CSV中的列

Csv.reader()方法所返回的每一行都是一個列表，因此我們可以按照索引的方式獲取每一列的值。

import csv

with open('example.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row[0], row[1], row[2])

結果輸出為:

1 John Doe
2 Jane Doe
3 Bob Smith

四、行定界符

當讀取CSV文件時，我們可以指定行定界符。默認情況下，行定界符為’\r’。

import csv

with open('example.csv', 'r', newline='') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

注意這裡的newline=”，這是為了防止將’\r’識別為新行符而加入的代碼。

五、自定義分隔符

當CSV文件的分隔符不是逗號時，我們可以使用delimiter來自定義分隔符。

import csv

with open('example.tsv', 'r') as tsvfile:
    reader = csv.reader(tsvfile, delimiter='\t')
    for row in reader:
        print(row)

這裡，我們使用’\t’作為分隔符來讀取TSV文件。

六、跳過標題行

有時，CSV文件的第一行是標題行，我們可以使用next()函數來跳過標題行。

import csv

with open('example.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    next(reader) # 跳過標題行
    for row in reader:
        print(row)

結果輸出為:

['1', 'John', 'Doe', 'example@gmail.com', '']
['2', 'Jane', 'Doe', 'jane@example.com', '']
['3', 'Bob', 'Smith', 'bob@example.com', '']