Импорт значений .csv в виде единого списка в Python

#python #pandas

#python #Список #csv-файл

Вопрос:

У меня есть csv файл, в котором есть один столбец записей, который выглядит следующим образом:

 test1
test2
test3
...
 

Я импортировал их в список Python, используя это:

 import csv

results = []
with open('test.csv', newline='') as inputfile:
    for row in csv.reader(inputfile):
        results.append(row)

print(results)
 

Однако это сработало нормально, оно выводит их следующим образом:

 [['test1'],['test2'],['test3']]
 

Как бы мне настроить свой код, чтобы они печатались в одном списке, например:

 ['test1','test2','test3']
 

и является ли один выход более предпочтительным, чем другой? Я понимаю, что это зависит от варианта использования. Мне просто любопытно, как другие работают с импортированными списками через файл .csv и какое преимущество один будет иметь перед другим.

Спасибо!

Комментарии:

1. У вас нет csv… у вас просто есть … текстовое сообщение…

2. это правда. возможно, в этом конкретном примере файл .txt был бы более подходящим.

Ответ №1:

Попробуйте следующий код

 import csv

results = []
with open('test.csv', newline='') as inputfile:
    for row in csv.reader(inputfile):
        results.append(row[0])

print(results)
 

Комментарии:

1. это хорошо работает. И если ваш CSV-файл содержит несколько столбцов, но вы хотите использовать только один из столбцов в качестве списка, просто измените номер индекса в (row[0]) нем.