python - 是否有用于此的 Python 函数？

Question

好的，所以我手动编写了一个函数来执行此操作，但我想知道是否有内置的 python/pandas/numpy/... 函数。基本上我想要的是

data_col = data.loc[data['col3'] == 'a']
data_final = data_col['col2']

但我希望它具有 col3 的所有值。所以它来自:

至:
请注意 col1 中的值是如何不存在的。如果没有您能想到的功能可以做这样的事情，请不要担心制作一个。我已经完成了它，它符合我的需求。只是好奇，我还没有完成我在学校的所有类(class)，所以我不是很熟悉所有的功能。

Answer 1

代码

df = df.pivot_table(
    'col2', columns='col3', aggfunc=(lambda x:x.to_list())
).apply(pd.Series.explode).rename_axis(None, axis=1).reset_index(drop=True))

输出

    a   b   c   d   e
0   6   7   8   9   10
1   10  9   8   7   6

说明

通过使用数据透视表，我们可以根据 col3 的组列出 col2 的所有元素，然后我们可以分解列表。

最后使用 rename_axis 和 reset_index 来根据需要转换索引。