以下是一个示例代码,演示了如何按标签选择并将数据写入CSV文件。
import pandas as pd
# 创建一个示例数据集
data = {'Name': ['Tom', 'Nick', 'John', 'Alice', 'Bob'],
'Age': [24, 32, 18, 45, 27],
'Gender': ['Male', 'Male', 'Male', 'Female', 'Male'],
'City': ['New York', 'Paris', 'London', 'Tokyo', 'Sydney']}
df = pd.DataFrame(data)
# 按标签选择数据
selected_data = df.loc[df['Age'] > 25]
# 将选择的数据写入CSV文件
selected_data.to_csv('selected_data.csv', index=False)
上述代码首先创建了一个示例数据集,包含了姓名、年龄、性别和城市四个字段。然后使用loc函数选择了年龄大于25的数据,将结果保存在selected_data变量中。最后,使用to_csv函数将选择的数据写入名为selected_data.csv的CSV文件中,其中index=False参数表示不写入行索引。
上一篇:按标签确定每列的计数