如何使用 pandas 在单行列中添加多个数据

Question

我有一个 DataFrame，它有两列名为 Brand 和 con。

import pandas as pd

cars = {'Brand': ['Honda Civic','Toyota Corolla'],
        'Con': [['India','Srilanka'],['India']]
        }

df = pd.DataFrame(cars, columns = ['Brand', 'Con'])

print (df)

这给了我

            Brand                Con
0     Honda Civic  [India, Srilanka]
1  Toyota Corolla            [India]

我想将 dataframe 转换为此 output

                           Brand       Con
0  [Honda Civic, Toyota Corolla]     India
1                  [Honda Civic]  Srilanka

先感谢您：）

Answer 1

利用：

df.explode('Con').groupby('Con')['Brand'].agg(list).reset_index().reindex(df.columns, axis=1)

编辑：

cars = {'Brand': ['Honda Civic','Toyota Corolla'],
        'Con': [['India','Srilanka'],['India']]
        }

df = pd.DataFrame(cars, columns = ['Brand', 'Con'])

print (df)

from itertools import chain

df1 = pd.DataFrame({
    'Con' : list(chain.from_iterable(df['Con'].tolist())), 
    'Brand' : df['Brand'].repeat(df['Con'].str.len())
})

df2 = df1.groupby('Con')['Brand'].agg(list).reset_index().reindex(df.columns, axis=1)
print(df2)
                           Brand       Con
0  [Honda Civic, Toyota Corolla]     India
1                  [Honda Civic]  Srilanka