通过设置固定长度将列表转换为字典

Question

我有一个列表data=['CDs', 1, 'J12345','Rainbow', None, 'Styles', 2, 'J12345', 'Rainbow', None, 'Folk', 3, 'J12345', 'Rainbow', None]我想把它转换成一个列数固定的数据帧。你知道吗

预期结果如下：

category |  num | series  |  title |  brand |
'CDs'       1     J12345   Rainbow   None 
'Styles'    2     J12345   Rainbow   None 
'Folk'      3     J12345   Rainbow   None 

Answer 1

设置

num_cols = 5
cols = ['category', 'num', 'series', 'title', 'brand']

---

numpy.reshape

d = np.reshape(data, (-1, num_cols))  
pd.DataFrame(d, columns=cols)

  category num  series    title brand
0      CDs   1  J12345  Rainbow  None
1   Styles   2  J12345  Rainbow  None
2     Folk   3  J12345  Rainbow  None

Answer 2

你可以做：

import pandas as pd

data = ['CDs', 1, 'J12345','Rainbow', None,
        'Styles', 2, 'J12345', 'Rainbow', None,
        'Folk', 3, 'J12345', 'Rainbow', None]

df = pd.DataFrame([data[i:i+5] for i in range(0, len(data), 5)], columns=['category', 'num', 'series',  'title', 'brand'])
print(df)

输出

  category  num  series    title brand
0      CDs    1  J12345  Rainbow  None
1   Styles    2  J12345  Rainbow  None
2     Folk    3  J12345  Rainbow  None

声明：

[data[i:i+5] for i in range(0, len(data), 5)]

创建以下列表：

[['CDs', 1, 'J12345', 'Rainbow', None], ['Styles', 2, 'J12345', 'Rainbow', None], ['Folk', 3, 'J12345', 'Rainbow', None]]

以上可以直接传给熊猫。你知道吗

Answer 3

和@user3483203的想法差不多

a = np.array(data)
nrows = 3
ncols = a.shape[0]//nrows
df = pd.DataFrame(a.reshape(nrows, ncols), columns=['category', 'num', 'series', 'title', 'brand'] )

    category    num series  title   brand
0   CDs         1   J12345  Rainbow None
1   Styles      2   J12345  Rainbow None
2   Folk        3   J12345  Rainbow None