Impor python csv ke daftar

Saya memiliki file CSV dengan sekitar 2000 catatan.

Setiap catatan memiliki string, dan kategori untuk itu:

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

Saya perlu membaca file ini menjadi daftar yang terlihat seperti ini:

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

Bagaimana cara mengimpor CSV ini ke daftar yang saya perlukan menggunakan Python?

Menggunakan modul csv :

import csv

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

print(data)

Keluaran:

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

Jika Anda membutuhkan tupel:

import csv

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

print(data)

Keluaran:

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

Old Python 2 menjawab, juga menggunakan csvmodul:

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']]

Diperbarui untuk Python 3 :

import csv

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

print(your_list)

Keluaran:

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

Panda cukup bagus dalam menangani data. Berikut ini satu contoh cara menggunakannya:

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()

Satu keuntungan besar adalah bahwa panda berurusan secara otomatis dengan baris tajuk.

Jika Anda belum pernah mendengar tentang Seaborn , saya sarankan untuk melihatnya.

Lihat juga: Bagaimana cara saya membaca dan menulis file CSV dengan Python?

Panda # 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')

Isi df adalah:

     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

Isi dari dikt adalah

[{'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}]

Panda # 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')]

Isi dari listsadalah:

[['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]]

Pembaruan untuk 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)

Keluaran:

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

Jika csvfile adalah objek file, itu harus dibuka bersama newline=''.
modul csv

Jika Anda yakin tidak ada koma di masukan Anda, selain untuk memisahkan kategori, Anda dapat membaca file baris demi baris dan membagi pada ,, kemudian dorong hasil untukList

Yang mengatakan, sepertinya Anda sedang melihat file CSV, jadi Anda dapat mempertimbangkan menggunakan modul untuk itu

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

Seperti yang sudah dikatakan di komentar Anda dapat menggunakan csvperpustakaan dengan python. csv berarti nilai yang dipisahkan koma, yang tampaknya persis dengan kasus Anda: label dan nilai yang dipisahkan oleh koma.

Menjadi kategori dan tipe nilai saya lebih suka menggunakan tipe kamus daripada daftar tupel.

Pokoknya dalam kode di bawah ini saya menunjukkan kedua cara: dadalah kamus dan ldaftar tupel.

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)

Sebuah loop sederhana sudah cukup:

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

print lines

Sayangnya saya menemukan tidak ada jawaban yang ada yang sangat memuaskan.

Berikut ini adalah solusi Python 3 yang mudah dan lengkap, menggunakan modul csv .

import csv

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

print(rows)

Perhatikan skipinitialspace=True argumennya. Ini diperlukan karena, sayangnya, CSV OP berisi spasi putih setelah setiap koma.

Keluaran:

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

Memperluas persyaratan Anda sedikit dan dengan anggapan Anda tidak peduli dengan urutan garis dan ingin membuat mereka dikelompokkan dalam kategori, solusi berikut dapat bekerja untuk Anda:

>>> 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']})

Dengan cara ini Anda mendapatkan semua baris relevan yang tersedia di kamus di bawah kunci menjadi kategori.

Berikut ini cara termudah dalam Python 3.x untuk mengimpor CSV ke array multidimensi, dan hanya 4 baris kode tanpa mengimpor apa pun!

#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)

Berikutnya adalah sepotong kode yang menggunakan modul csv tetapi mengekstrak konten file.csv ke daftar dicts menggunakan baris pertama yang merupakan header dari tabel 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