Объединение Pandas данных с Numpy массивами

Question

У меня есть массив массивов, которые я хочу объединить с фреймом данных.

arrays=[np.array(i) for i in [[1,2],[5,6,7],[]]] #let me illustrate the arrays like this

df=pd.DataFrame({'Col':['x','y','z']})

Каждый элемент массива соответствует строке в df. Это мой желаемый вывод:

Вот способ, которым вы могли бы продублировать строки df для размещения входящих в массив элементов:

df.loc[df.index.repeat([max(1,len(i)) for i in arrays])]

Спасибо

Answer 1

Вы можете сделать следующее:

import numpy as np
import pandas as pd
from itertools import product

arrays = [np.array(i) for i in [[1, 2], [5, 6, 7], []]]
df = pd.DataFrame({'Col': ['x', 'y', 'z']})

# this creates a mesh (cross-product) Dataframe
mesh = pd.DataFrame([pair for co in zip(df['Col'], arrays) for pair in product(*co)],
                    columns=['Col', 'n'])

# merge with the original Dataframe
result = df.merge(mesh, on='Col', how='left').fillna(0)

Вывод

  Col    n
0   x  1.0
1   x  2.0
2   y  5.0
3   y  6.0
4   y  7.0
5   z  0.0

Answer 2

df=pd.DataFrame({'Col':['x','y','z']})

arrays=[i for i in [[1,2],[5,6,7],[]]] #let me illustrate the arrays like this

df['newcol']=arrays


df.explode('newcol')