Python nhập csv vào danh sách

Tôi có một tệp CSV với khoảng 2000 hồ sơ.

Mỗi bản ghi có một chuỗi và một danh mục cho nó:

This is the first line,Line1
This is the second line,Line2
This is the third line,Line3

Tôi cần đọc tệp này vào một danh sách giống như thế này:

data = [('This is the first line', 'Line1'),
        ('This is the second line', 'Line2'),
        ('This is the third line', 'Line3')]

Làm cách nào để nhập CSV này vào danh sách tôi cần bằng Python?

Sử dụng mô-đun csv :

import csv

with open('file.csv', newline='') as f:
    reader = csv.reader(f)
    data = list(reader)

print(data)

Đầu ra:

[['This is the first line', 'Line1'], ['This is the second line', 'Line2'], ['This is the third line', 'Line3']]

Nếu bạn cần bộ dữ liệu:

import csv

with open('file.csv', newline='') as f:
    reader = csv.reader(f)
    data = [tuple(row) for row in reader]

print(data)

Đầu ra:

[('This is the first line', 'Line1'), ('This is the second line', 'Line2'), ('This is the third line', 'Line3')]

Câu trả lời cũ của Python 2, cũng sử dụng csvmô-đun:

import csv
with open('file.csv', 'rb') as f:
    reader = csv.reader(f)
    your_list = list(reader)

print your_list
# [['This is the first line', 'Line1'],
#  ['This is the second line', 'Line2'],
#  ['This is the third line', 'Line3']]

Đã cập nhật cho Python 3 :

import csv

with open('file.csv', newline='') as f:
    reader = csv.reader(f)
    your_list = list(reader)

print(your_list)

Đầu ra:

[['This is the first line', 'Line1'], ['This is the second line', 'Line2'], ['This is the third line', 'Line3']]

Gấu trúc khá giỏi trong việc xử lý dữ liệu. Đây là một ví dụ về cách sử dụng nó:

import pandas as pd

# Read the CSV into a pandas data frame (df)
#   With a df you can do many things
#   most important: visualize data with Seaborn
df = pd.read_csv('filename.csv', delimiter=',')

# Or export it in many ways, e.g. a list of tuples
tuples = [tuple(x) for x in df.values]

# or export it as a list of dicts
dicts = df.to_dict().values()

Một lợi thế lớn là gấu trúc tự động xử lý các hàng tiêu đề.

Nếu bạn chưa từng nghe về Seaborn , tôi khuyên bạn nên xem qua.

Xem thêm: Làm cách nào để đọc và ghi tệp CSV bằng Python?

Gấu trúc # 2

import pandas as pd

# Get data - reading the CSV file
import mpu.pd
df = mpu.pd.example_df()

# Convert
dicts = df.to_dict('records')

Nội dung của df là:

     country   population population_time    EUR
0    Germany   82521653.0      2016-12-01   True
1     France   66991000.0      2017-01-01   True
2  Indonesia  255461700.0      2017-01-01  False
3    Ireland    4761865.0             NaT   True
4      Spain   46549045.0      2017-06-01   True
5    Vatican          NaN             NaT   True

Nội dung của dicts là

[{'country': 'Germany', 'population': 82521653.0, 'population_time': Timestamp('2016-12-01 00:00:00'), 'EUR': True},
 {'country': 'France', 'population': 66991000.0, 'population_time': Timestamp('2017-01-01 00:00:00'), 'EUR': True},
 {'country': 'Indonesia', 'population': 255461700.0, 'population_time': Timestamp('2017-01-01 00:00:00'), 'EUR': False},
 {'country': 'Ireland', 'population': 4761865.0, 'population_time': NaT, 'EUR': True},
 {'country': 'Spain', 'population': 46549045.0, 'population_time': Timestamp('2017-06-01 00:00:00'), 'EUR': True},
 {'country': 'Vatican', 'population': nan, 'population_time': NaT, 'EUR': True}]

Gấu trúc # 3

import pandas as pd

# Get data - reading the CSV file
import mpu.pd
df = mpu.pd.example_df()

# Convert
lists = [[row[col] for col in df.columns] for row in df.to_dict('records')]

Nội dung của listslà:

[['Germany', 82521653.0, Timestamp('2016-12-01 00:00:00'), True],
 ['France', 66991000.0, Timestamp('2017-01-01 00:00:00'), True],
 ['Indonesia', 255461700.0, Timestamp('2017-01-01 00:00:00'), False],
 ['Ireland', 4761865.0, NaT, True],
 ['Spain', 46549045.0, Timestamp('2017-06-01 00:00:00'), True],
 ['Vatican', nan, NaT, True]]

Cập nhật cho Python3:

import csv
from pprint import pprint

with open('text.csv', newline='') as file:
    reader = csv.reader(file)
    res = list(map(tuple, reader))

pprint(res)

Đầu ra:

[('This is the first line', ' Line1'),
 ('This is the second line', ' Line2'),
 ('This is the third line', ' Line3')]

Nếu csvfile là một đối tượng tập tin, nó sẽ được mở bằng newline=''.
mô đun csv

Nếu bạn chắc chắn không có dấu phẩy trong đầu vào của bạn, trừ để tách các hạng mục, bạn có thể đọc những dòng tập tin bằng cách dòng và chia trên ,, sau đó đẩy các kết quả đểList

Điều đó nói rằng, có vẻ như bạn đang xem một tệp CSV, vì vậy bạn có thể cân nhắc sử dụng các mô-đun cho nó

result = []
for line in text.splitlines():
    result.append(tuple(line.split(",")))

Như đã nói trong các ý kiến, bạn có thể sử dụng csvthư viện trong python. csv có nghĩa là các giá trị được phân tách bằng dấu phẩy có vẻ chính xác là trường hợp của bạn: nhãn và giá trị được phân tách bằng dấu phẩy.

Là một loại và loại giá trị, tôi muốn sử dụng một loại từ điển thay vì một danh sách các bộ dữ liệu.

Dù sao trong đoạn mã dưới đây tôi chỉ ra cả hai cách: dlà từ điển và llà danh sách các bộ dữ liệu.

import csv

file_name = "test.txt"
try:
    csvfile = open(file_name, 'rt')
except:
    print("File not found")
csvReader = csv.reader(csvfile, delimiter=",")
d = dict()
l =  list()
for row in csvReader:
    d[row[1]] = row[0]
    l.append((row[0], row[1]))
print(d)
print(l)

Một vòng lặp đơn giản sẽ đủ:

lines = []
with open('test.txt', 'r') as f:
    for line in f.readlines():
        l,name = line.strip().split(',')
        lines.append((l,name))

print lines

Thật không may, tôi không tìm thấy câu trả lời nào đặc biệt thỏa mãn.

Đây là một giải pháp Python 3 đơn giản và đầy đủ, sử dụng mô-đun csv .

import csv

with open('../resources/temp_in.csv', newline='') as f:
    reader = csv.reader(f, skipinitialspace=True)
    rows = list(reader)

print(rows)

Chú ý skipinitialspace=Truelập luận. Điều này là cần thiết vì thật không may, CSV của OP chứa khoảng trắng sau mỗi dấu phẩy.

Đầu ra:

[['This is the first line', 'Line1'], ['This is the second line', 'Line2'], ['This is the third line', 'Line3']]

Mở rộng yêu cầu của bạn một chút và giả sử bạn không quan tâm đến thứ tự các dòng và muốn phân nhóm chúng theo các danh mục, giải pháp sau có thể phù hợp với bạn:

>>> fname = "lines.txt"
>>> from collections import defaultdict
>>> dct = defaultdict(list)
>>> with open(fname) as f:
...     for line in f:
...         text, cat = line.rstrip("\n").split(",", 1)
...         dct[cat].append(text)
...
>>> dct
defaultdict(<type 'list'>, {' CatA': ['This is the first line', 'This is the another line'], ' CatC': ['This is the third line'], ' CatB': ['This is the second line', 'This is the last line']})

Bằng cách này, bạn có được tất cả các dòng có liên quan có sẵn trong từ điển dưới khóa là thể loại.

Đây là cách dễ nhất trong Python 3.x để nhập CSV vào một mảng nhiều chiều và chỉ có 4 dòng mã mà không cần nhập bất cứ thứ gì!

#pull a CSV into a multidimensional array in 4 lines!

L=[]                            #Create an empty list for the main array
for line in open('log.txt'):    #Open the file and read all the lines
    x=line.rstrip()             #Strip the \n from each line
    L.append(x.split(','))      #Split each line into a list and add it to the
                                #Multidimensional array
print(L)

Tiếp theo là một đoạn mã sử dụng mô-đun csv nhưng trích xuất nội dung file.csv vào danh sách các ký tự sử dụng dòng đầu tiên là tiêu đề của bảng csv

import csv
def csv2dicts(filename):
  with open(filename, 'rb') as f:
    reader = csv.reader(f)
    lines = list(reader)
    if len(lines) < 2: return None
    names = lines[0]
    if len(names) < 1: return None
    dicts = []
    for values in lines[1:]:
      if len(values) != len(names): return None
      d = {}
      for i,_ in enumerate(names):
        d[names[i]] = values[i]
      dicts.append(d)
    return dicts
  return None

if __name__ == '__main__':
  your_list = csv2dicts('file.csv')
  print your_list