如何在Python中创建与excel类似的计算字段？

Question

我希望将透视表从Excel迁移到python，以便使用可视化和其他方法。我在excel中使用了两个计算字段，所以我想知道是否可以使用Pandas的类似想法？谢谢。

Answer 1

不确定你的数据是什么样子，但这是绝对有可能的熊猫。

下面是一个例子：

# example dataframe
df = pd.DataFrame({'age': [17, 23, 4, 27],
                    'name': ['John', 'Mark', 'Alice', 'Alice']})

Output1

    age name
0   17  John
1   23  Mark
2   4   Alice
3   27  Alice

用np.where方法创建计算字段

这种方法背后的逻辑：np.where(condition, true value, false value)

找到更多的这里

df['adult_indicator'] = np.where(df.age >= 18, 1, 0)

Output2

    age name    adult_indicator
0   17  John    0
1   23  Mark    1
2   4   Alice   0
3   27  Alice   1

从pivot模块应用pandas方法

df.pivot(index='name', columns='age', values='adult_indicator')

Output3

    age 4   17  23  27
name                
Alice   0.0 NaN NaN 1.0
John    NaN 0.0 NaN NaN
Mark    NaN NaN 1.0 NaN