在 Pandas 中将分类列转换为 int - Python

在 Pandas 中，将分类列转换为整数（int）是常见任务之一。这个过程可以通过使用 cat.codes 方法简便地完成。

如何将分类列转换为整数?

下面是示例代码：

import pandas as pd

# 创建示例数据帧
df = pd.DataFrame({'col1': ['A', 'B', 'C', 'A', 'B', 'C'],
                   'col2': ['X', 'Y', 'Z', 'X', 'Y', 'Z']})

# 将 'col1' 列转换成整数型 (int)
df['col1_int'] = df['col1'].astype('category').cat.codes

print(df)

上述代码执行后，输出结果如下:

  col1 col2  col1_int
0    A    X         0
1    B    Y         1
2    C    Z         2
3    A    X         0
4    B    Y         1
5    C    Z         2

在上述示例中，我们通过 astype('category').cat.codes 方法将 col1 列转换为整数，然后将结果存储到新列 col1_int 中。通过上述代码，我们成功地将分类列转换为整数型。

总结

分类列转换为整数是一个常见任务，Pandas 在这方面提供了很好的支持。使用 Pandas 中的 cat.codes 方法，我们可以轻松地将分类列转换为整数型。