Скажем, у меня есть словарь, в котором 10 пар ключ-значение. Каждая запись содержит массив NumPy. Однако, длина массива-это не то же самое для всех из них.
Как я могу создать таблицу данных, где каждый столбец имеет другую запись?
Когда я попробовать:
pd.DataFrame(my_dict)
Я получаю:
ValueError: arrays must all be the same length
Любой способ избавиться от этого? Я счастлива, что у панд использовать Нана
, чтобы дополнить эти столбцы для коротких записей.
В Python 3.х:
In [6]: d = dict( A = np.array([1,2]), B = np.array([1,2,3,4]) )
In [7]: DataFrame(dict([ (k,Series(v)) for k,v in d.items() ]))
Out[7]:
A B
0 1 1
1 2 2
2 NaN 3
3 NaN 4
В Python 2.х:
заменить д. элементы()с
д.iteritems()`.
Здесь'ы простой способ сделать это:
In[20]: my_dict = dict( A = np.array([1,2]), B = np.array([1,2,3,4]) )
In[21]: df = pd.DataFrame.from_dict(my_dict, orient='index')
In[22]: df
Out[22]:
0 1 2 3
A 1 2 NaN NaN
B 1 2 3 4
In[23]: df.transpose()
Out[23]:
A B
0 1 1
1 2 2
2 NaN 3
3 NaN 4
Способ подчистить свой синтаксис, но до сих пор по сути то же самое, что эти другие ответы ниже:
>>> mydict = {'one': [1,2,3], 2: [4,5,6,7], 3: 8}
>>> dict_df = pd.DataFrame({ key:pd.Series(value) for key, value in mydict.items() })
>>> dict_df
one 2 3
0 1.0 4 8.0
1 2.0 5 NaN
2 3.0 6 NaN
3 NaN 7 NaN
Существует схожий синтаксис для списков, слишком:
>>> mylist = [ [1,2,3], [4,5], 6 ]
>>> list_df = pd.DataFrame([ pd.Series(value) for value in mylist ])
>>> list_df
0 1 2
0 1.0 2.0 3.0
1 4.0 5.0 NaN
2 6.0 NaN NaN
Другой синтаксис для списков:
>>> mylist = [ [1,2,3], [4,5], 6 ]
>>> list_df = pd.DataFrame({ i:pd.Series(value) for i, value in enumerate(mylist) })
>>> list_df
0 1 2
0 1 4.0 6.0
1 2 5.0 NaN
2 3 NaN NaN
Во всех этих случаях, вы должны быть осторожны, чтобы проверить, какой тип данных панды
собирается угадать в вашей колонке. Столбцы, содержащие любой Нана
(отсутствующие) значения будут преобразованы в поплавок, например.
Хотя это не прямо ответить на ОП'ы вопрос. Я нашел, что это отличное решение для моего случая, когда я имел неравные массивы и я'd хотелось поделиться:
In [31]: d = {'one' : Series([1., 2., 3.], index=['a', 'b', 'c']),
....: 'two' : Series([1., 2., 3., 4.], index=['a', 'b', 'c', 'd'])}
....:
In [32]: df = DataFrame(d)
In [33]: df
Out[33]:
one two
a 1 1
b 2 2
c 3 3
d NaN 4
Вы также можете использовать ПД.функция concat
вдоль оси=1
Список `ПД.Объекты серии:
import pandas as pd, numpy as np
d = {'A': np.array([1,2]), 'B': np.array([1,2,3,4])}
res = pd.concat([pd.Series(v, name=k) for k, v in d.items()], axis=1)
print(res)
A B
0 1.0 1
1 2.0 2
2 NaN 3
3 NaN 4
Обе следующие строки прекрасно работать :
pd.DataFrame.from_dict(df, orient='index').transpose() #A
pd.DataFrame(dict([ (k,pd.Series(v)) for k,v in df.items() ])) #B (Better)
Но с %раз все на Jupyter, я'ве получил соотношение 4х скорости для Б против С, что весьма впечатляет, особенно при работе с большим набором данных (в основном с большим количеством колонн/характеристики).
Если вы Don'т хочу, чтобы это показать Нана
и у вас есть две длины, добавив 'космос' в каждой из оставшихся клеток также будет работать.
import pandas
long = [6, 4, 7, 3]
short = [5, 6]
for n in range(len(long) - len(short)):
short.append(' ')
df = pd.DataFrame({'A':long, 'B':short}]
# Make sure Excel file exists in the working directory
datatoexcel = pd.ExcelWriter('example1.xlsx',engine = 'xlsxwriter')
df.to_excel(datatoexcel,sheet_name = 'Sheet1')
datatoexcel.save()
A B
0 6 5
1 4 6
2 7
3 3
Если у вас больше чем 2 длины записи, желательно сделать функцию, которая использует подобный метод.