Pandas DF 到嵌套列表

Question

我有一個 dataframe (df)，我想將它轉換為嵌套列表。

df=pd.DataFrame({'Number':[1,2,3,4,5, 6],
             'Name':['A', 'B', 'C', 'D', 'E', 'F'],
             'Value': [223, 124, 434, 514, 821, 110]})

我的預期結果是一個嵌套列表。 嵌套內的第一個列表從第一列的 df 的前 3 行獲取值。 第二個是第二列的前 3 行，第三個是第三列的前 3 行。 之后，我想添加剩余 3 行的列表。

[[1, 2, 3], 
 ['A', 'B', 'C'], 
 [223, 124, 434]
 [4, 5, 6], 
['D', 'E', 'F'],
[514, 821, 110]]

我做了一個 for 循環並在每個系列上調用了 tolist() 。 然后我從列表中的一列中獲取所有值。 我如何從下面的結果到上面的預期結果 go？

col=df.columns

lst=[]
for i in col:
        temp = df[i].tolist()
        temp
        lst.append(temp)

結果（第一）：

[[1, 2, 3, 4, 5, 6],
 ['A', 'B', 'C', 'D', 'E', 'F'],
 [223, 124, 434, 514, 821, 110]]

Answer 1

使用.values和一些 numpy 切片

v = df.values.T
v[:,:3].tolist() + v[:,3:].tolist()

output

[[1, 2, 3],
 ['A', 'B', 'C'],
 [223, 124, 434],
 [4, 5, 6],
 ['D', 'E', 'F'],
 [514, 821, 110]]

Answer 2

嘗試：

lst = df.set_index(df.index // 3).groupby(level=0).agg(list) \
        .to_numpy().ravel().tolist()
print(lst)

# Output
[[1, 2, 3],
 ['A', 'B', 'C'],
 [223, 124, 434],
 [4, 5, 6],
 ['D', 'E', 'F'],
 [514, 821, 110]]

Answer 3

這是一個從 3 個列表開始的示例，你得到的列表是.tolist()

a = [1, 2, 3, 4, 5, 6, 4]
b = ['A', 'B', 'C', 'D', 'E', 'F']
c = [223, 124, 434, 514, 821, 110]

res = []

for i in range(len(a) // 3):
  res.append(a[i * 3:(i * 3) + 3])
  res.append(b[i * 3:(i * 3) + 3])
  res.append(c[i * 3:(i * 3) + 3])

結果是

[[1, 2, 3], ['A', 'B', 'C'], [223, 124, 434], [4, 5, 6], ['D', 'E', 'F'], [514, 821, 110]]

Answer 4

import pandas as pd
df=pd.DataFrame({
    'Number':[1,2,3,4,5, 6],
    'Name':['A', 'B', 'C', 'D', 'E', 'F'],
    'Value': [223, 124, 434, 514, 821, 110]
})

# convert df into slices of 3

final_list = []
for i in range(0, len(df), 3):
    final_list.append(
        df.iloc[i:i+3]['Number'].to_list())
    final_list.append(
        df.iloc[i:i+3]['Name'].to_list())
    final_list.append(
        df.iloc[i:i+3]['Value'].to_list())
    

print(final_list)

output

[[1, 2, 3], ['A', 'B', 'C'], [223, 124, 434], [4, 5, 6], ['D', 'E', 'F'], [514, 821, 110]]

Answer 5

我認為您只想將列表（列）划分為大小為 n 的列表。 您可以更改 n 的值，以更改子列表大小。

lst=[]
n=3
for i in col:
        temp = df[i].tolist()
        for i in range(0,len(temp),n):
            lst.append(temp[i:i+n])