根据pandas中的条件，将列名列表作为新列返回
在pandas中，可以使用条件来选择数据，并将选择的列名列表作为新列返回。具体步骤如下：
- 导入pandas库：首先需要导入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)- 使用条件选择数据：使用条件选择数据可以使用pandas的条件语句，例如，选择年龄大于30的数据可以使用以下代码：
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元无门槛券
手把手带您无忧上云
相关资讯
热门标签
活动推荐
腾讯云开发者
