在pandas中,可以使用条件来选择数据,并将选择的列名列表作为新列返回。具体步骤如下:
import pandas as pd
data = {'Name': ['John', 'Emma', 'Mike', 'Emily'],
'Age': [25, 30, 35, 28],
'Gender': ['Male', 'Female', 'Male', 'Female']}
df = pd.DataFrame(data)
condition = df['Age'] > 30
selected_data = df[condition]
selected_columns = ['Name', 'Age']
new_df = df[selected_columns]
完整的代码示例如下:
import pandas as pd
data = {'Name': ['John', 'Emma', 'Mike', 'Emily'],
'Age': [25, 30, 35, 28],
'Gender': ['Male', 'Female', 'Male', 'Female']}
df = pd.DataFrame(data)
condition = df['Age'] > 30
selected_data = df[condition]
selected_columns = ['Name', 'Age']
new_df = df[selected_columns]
在上述代码中,根据条件选择了年龄大于30的数据,并将选择的列名列表['Name', 'Age']作为新列返回。
领取专属 10元无门槛券
手把手带您无忧上云