python - 如何将列表中的元素一个一个地附加到数据框中的嵌套列表中

Question

import pandas as pd

d = {'A': [1,2,3,4], 'B': [[[1,2],[2,3]],[[3,4],[2,5]],[[5,6],[5,6],[5,6]],[7,8]]}

df = pd.DataFrame(data=d)

C = [1,2,3,4,5,6,7,8]

我有一个 pandas 数据框，我想将 C 列表的每个元素附加到 B 的每个嵌套列表中，维护结构，以便生成的数据框是:

'A': [1,2,3,4]
'B': [[[1,2,1],[2,3,2]],[[3,4,3],[2,5,4]],[[5,6,5],[5,6,6],[5,6,7]],[7,8,8]]

Answer 1

Mybe 有一个更优雅的解决方案，但是这个可行:-)

for i in d['B']:
    for j in i:
        if (isinstance(j, list)):
            j.append(C.pop(0))
        else:
            i.append(C.pop(0))
            break

基于timgebs评论的更高效的解决方案(谢谢!):

f = iter(C)
for i in d['B']:
    for j in i:
        if (isinstance(j, list)):
            j.append(next(f))
        else:
            i.append(next(f))
            break