是否同时执行逻辑操作和添加dataframe的新列？

Question

因为我是新来的，正在学习python，所以我想在数据帧中收集与逻辑操作相对应的特定数据，并为其添加一个标签，然而，这需要在许多代码行中执行

例如：

df = df[(df['this_col'] >= 10) & (df['anth_col'] < 100)]
result_df = df.copy()
result_df['label'] = 'medium'

我真的很想知道是否有一种方法可以在一行代码中执行，而无需应用函数。如果它不能在一行中执行，为什么会这样

干杯

Answer 1

query始终返回一个副本

result_df = df.query("this_col >= 10 and anth_col < 100").assign(label='medium')

假设您的列在python中可以作为有效的标识符名称传递，那么就可以了