Выберите все столбцы без первых 2 по позициям на DataFrame.iloc
, затем измените их на DataFrame.stack
, повторите MultiIndex
на Index.repeat
и агрегат list
s:
s = df.iloc[:, 2:].stack()
df['Products'] = s[s.index.repeat(s)].reset_index().groupby(['level_0'])['level_1'].agg(list)
print (df)
Day Id Banana Apple Products
0 2020-01-01 1 1.0 1 [Banana, Apple]
1 2020-01-02 1 NaN 2 [Apple, Apple]
2 2020-01-03 2 2.0 2 [Banana, Banana, Apple, Apple]
Или используйте пользовательскую функцию с повторяющимися columns
именами без пропущенных значений:
def f(x):
s = x.dropna()
return s.index.repeat(s).tolist()
df['Products'] = df.iloc[:, 2:].apply(f, axis=1)
print (df)
Day Id Banana Apple Products
0 2020-01-01 1 1.0 1 [Banana, Apple]
1 2020-01-02 1 NaN 2 [Apple, Apple]
2 2020-01-03 2 2.0 2 [Banana, Banana, Apple, Apple]