编写 Python 代码将给定的系列转换为虚拟变量,并删除任何 NaN 值(如果存在)

pythonpandasserver side programmingprogramming

假设您有一个系列,转换为虚拟变量的结果是,

   Female Male
0    0    1
1    1    0
2    0    1
3    1    0
4    0    1
5    0    0
6    1    0
7    1    0

为了解决这个问题,我们将遵循下面给出的步骤 −

解决方案

  • 创建一个包含"男性"和"女性"元素的列表,并将其分配到系列中。

  • 在系列中应用 get_dummies 函数并将 dummy_na 值设置为 False。它定义如下,

pd.get_dummies(series, dummy_na=False)

示例

让我们检查以下代码以获得更好的理解 −

import pandas as pd
import numpy as np
gender = ['Male','Female','Male','Female','Male',np.nan,'Female','Female',]
series = pd.Series(gender)
print("Series is:\n",series)
print("Dummy code is:\n", pd.get_dummies(series, dummy_na=False))

输出

Series is:
0    Male
1    Female
2    Male
3    Female
4    Male
5    NaN
6    Female
7    Female
dtype: object
Dummy code is:
   Female Male
0    0    1
1    1    0
2    0    1
3    1    0
4    0    1
5    0    0
6    1    0
7    1    0

相关文章