在Python-Pandas中用True和False替换包含’yes’和’no’值的列

  • Post category:Python

首先需要导入Pandas库:

import pandas as pd

然后读取包含’yes’和’no’值的csv文件,例如文件名为data.csv:

df = pd.read_csv('data.csv')

接下来,我们可以使用Pandas中的replace()方法来将’yes’替换为True,’no’替换为False:

df = df.replace({'yes':True, 'no':False})

其中replace()方法的参数是一个字典,键为需要被替换的值,值为替换后的值。

最后,我们可以将修改后的数据保存为csv文件:

df.to_csv('new_data.csv', index=False)

其中to_csv()方法的第一个参数为保存文件的文件名,第二个参数index=False表示不需要保存索引。

下面是完整的代码示例:

import pandas as pd

df = pd.read_csv('data.csv')

df = df.replace({'yes':True, 'no':False})

df.to_csv('new_data.csv', index=False)

注:本示例中假设原csv文件中只包含’yes’和’no’两个值,如果还包含其他值,则需要注意是否需要替换。