Прочитать CSV-файл в список - PullRequest
       0

Прочитать CSV-файл в список

0 голосов
/ 27 сентября 2018

Я хочу экспортировать вложенный список, затем импортировать его без потери формы списка.Список представляет собой вложенный список, каждый внутренний список имеет 10 элементов, 10000 внутренних списков составляют большой список.

Внутренние списки выглядят так: форма 1 x 10

    [array([  1, 2, 3, 4, 5
              6, 7, 8, 9, 10 ],
    array([ 21, 22, 23, 24, 25
           26, 27, 28, 29, 30 ]

Я экспортировал ее в CSV.В CSV, это становится так.Каждый внутренний список находится в одной ячейке

['[ 1  2  3  4  5\n 6  7  8  9  10]']
['[ 21  22  23  24  25\n 26  27  28  29  30]']

Затем я импортировал CSV в список.Тем не менее, форма становится 1 x 1, я хочу превратить ее в форму 1 x 10, в идеале такую ​​же, как исходный формат списка, который я экспортировал.как этого добиться?

Код, который я использовал для экспорта списка: (X - это список)

import csv
with open('X.csv', "w") as output:
    writer = csv.writer(output, lineterminator='\n')
    for val in X:
        writer.writerow([val])  

Код, который я использовал для импорта csv: (X - это список)

with open('X.csv', 'r') as f:
    reader = csv.reader(f)
    X = list(reader)

Спасибо заранее!

Ответы [ 2 ]

0 голосов
/ 27 сентября 2018

Я бы посоветовал вам, возможно, сохранить его как рассол или DataFrame, вместо этого вам не нужно слишком много возиться при чтении его обратно. Соленья сохраняются в представлении памяти, используемом python.

import pickle
pickle.dump(X, open('X.pkl', "wb"))
X_loaded = pickle.load(open('X.pkl', "rb"))
0 голосов
/ 27 сентября 2018

Попробуйте,

import pandas as pd
df = pd.read_csv('filename.csv')
list_ = df.values.tolist()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...