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

我该怎么办？

Answer 1

只需替换和重命名即可完成..

数据框:

>>> df
    Name  female  male
0   bayu       0     1
1  jason       0     1
2   anna       1     0

结果:

>>> df.replace({ 0: "male", 1: "Female"}).drop(columns={'male'}).rename(columns={'female': 'Gender'})
    Name  Gender
0   bayu    male
1  jason    male
2   anna  Female

将数据帧保存到 csv..

>>> df = df.replace({ 0: "male", 1: "Female"}).drop(columns {'male'}).rename(columns={'female': 'Gender'})
>>> df.to_csv('data.csv', index=False)

您的 CSV 将如下所示，以逗号分隔 ..

$ cat data.csv
Name,Gender
bayu,male
jason,male
anna,Female

或者，只需使用 idxmax 和 df.assign 将值一起分配回 nw 列。

>>> df.assign(Gender = df[['male','female']].idxmax(axis=1)).drop(columns={'male', 'female'})
    Name  Gender
0   bayu    male
1  jason    male
2   anna  female

2) 另一种创建函数然后使用 apply 方法的方法。

 def setval(col):
   col = str(col)
   if col.startswith('0'):
     return  'Male'
   if col.startswith('1'):
     return  'Female'


>>> df.assign(Gender = df[ 'female'].apply(setval)).drop({"female", "male"}, axis=1)
    Name  Gender
0   bayu    Male
1  jason    Male
2   anna  Female

3) 为了子孙后代..

您可以创建一个字典，然后将字典映射到其中一个列，使用 df.assign 进一步删除不需要的列，将新值分配给新列。

在这里，您可以随意使用map 或replace 方法..

>>> newVal = {0:'Male', 1: 'Female'}   # dictionary paired with key, values

a) 在下面的示例中，axis=1 表示使用较新方法的列。

>>> df.assign(Gender=df.female.map(newVal)).drop({'male', 'female'},axis=1)
    Name  Gender
0   bayu    Male
1  jason    Male
2   anna  Female

b) 而在下面的示例中，我们明确提到它应用于旧方法的列。

>>> df.assign(Gender=df.female.map(newVal)).drop(columns={'male', 'female'})
    Name  Gender
0   bayu    Male
1  jason    Male
2   anna  Female

用.replace

>>> df.assign(Gender=df.female.replace(newVal)).drop({'male', 'female'},axis=1)
    Name  Gender
0   bayu    Male
1  jason    Male
2   anna  Female