使用 Pandas 函数创建多个新列
在 Pandas 中,可以将单个函数应用于列并基于 Pandas 函数创建多个新列
考虑以下场景:
您有一个函数 extract_text_features,它接受一个文本列并返回 6 个输出列。虽然函数本身可以工作,但将输出分配给特定列会带来挑战。
使用 Zip 的解决方案:
一种有效的方法是使用 zip 函数来分配将值输出到多列:
<code class="python">def powers(x): return x, x**2, x**3, x**4, x**5, x**6 df = pd.DataFrame([[i] for i in range(10)], columns=['num']) df['p1'], df['p2'], df['p3'], df['p4'], df['p5'], df['p6'] = zip(*df['num'].map(powers))</code>
这里,map() 函数将幂应用于“num”列的每一行,并且 zip 将输出解压缩到各个列中。
这方法根据函数的输出值有效地生成多个新列。它也与旧版本的 Pandas 兼容,不像其他一些依赖 df.assign() 等新功能的解决方案。
以上是如何从 Pandas 中的单个函数输出创建多个新列?的详细内容。更多信息请关注PHP中文网其他相关文章!