python - 如何处理从文本到 CSV 的数据

Question

我有这样的文本文件 data.txt：

Name male  female
bayu 1 0
jonson 1 0
anna 0 1

我尝试使用 pandas python 转换这个文件

import pandas as pd
df = pd.read_fwf('data.txt')
df.to_csv('data.csv')

我想得到这样的结果data.csv：

Gender name
bayu male
jonson male
Anna  Female

我应该怎么办？

标签： pythonpython-3.xpandasdataframe

Answer 1

import numpy as np
df['Gender'] = np.where(df['male'].eq(1), 'male', 'female')
df = df.drop(columns={'male', 'female'})
df.to_csv('data.csv', index=False)

    Name    Gender
0   bayu    male
1   jonson  male
2   anna    female