Как преобразовать многомерный список в массив Numpy, а затем записать в CSV-файл - PullRequest
0 голосов
/ 02 июля 2019

Я выполняю запрос с использованием psycopg2 в Python. Результаты запроса сохраняются в списке. Я пытаюсь преобразовать этот список в массив Numpy, а затем записать в CSV-файл. Вот как я это сделал.

rows = rcursor.fetchall()
df = pd.DataFrame(np.array(rows), columns = rows("db1 db2 db3 db4 db5"))
df.to_csv('alldata.csv',sep=',')]

Но когда я это делаю, я получаю ошибку:

ValueError: Must pass 2-d input

Думаю, мне нужно применить .reshape, но количество строк огромно (например, 200000). Данные, извлекаемые из запроса в список, выглядят следующим образом.

RealDictRow([('db1', '0001'), ('db2', 002), ('db3', 003), ('db4', '004'), ('db5', 'Hello I worked on this so far but not happening. Call my number 245-456-7892)

Как я могу правильно написать это в csv, не получая ValueError: Должен пропустить 2-й ввод. Заранее спасибо!

1 Ответ

0 голосов
/ 03 июля 2019

Вы можете использовать pandas.read_sql_query напрямую, чтобы преобразовать SQL-запрос + соединение в фрейм данных. См. здесь , например.

...