pandas把字符串转化为数字|极客教程

pandas把字符串转化为数字

在数据处理过程中，我们经常会遇到需要把字符串转化为数字的情况。比如，有时候我们从数据库或者文件读取数据时，数据以字符串的形式存在，但是我们需要把这些数据转化为数字进行分析和计算。在Python中，使用pandas库可以很方便地实现字符串转化为数字的操作。本文将详细介绍使用pandas库将字符串转化为数字的方法。

1. 使用astype()方法

在pandas中，可以使用astype()方法将一列数据的数据类型进行转换。当数据以字符串形式存在时，可以使用astype()方法将字符串转化为数字。下面是一个简单的示例：

import pandas as pd

# 创建一个DataFrame
data = {'A': ['1', '2', '3', '4', '5'],
        'B': ['5', '4', '3', '2', '1']}
df = pd.DataFrame(data)

# 将'A'列的数据转化为整数类型
df['A'] = df['A'].astype(int)

# 将'B'列的数据转化为浮点数类型
df['B'] = df['B'].astype(float)

print(df.dtypes)
print(df)

运行以上代码，我们得到的输出如下：

A      int32
B    float64
dtype: object

   A    B
0  1  5.0
1  2  4.0
2  3  3.0
3  4  2.0
4  5  1.0

可以看到，通过使用astype()方法，我们成功将字符串转化为数字，并且数据类型也发生了相应的改变。

2. 使用to_numeric()函数

除了astype()方法，pandas还提供了一个to_numeric()函数，专门用于将数据转化为数字。这个函数还可以处理一些特殊情况，比如数据中包含了特殊字符等。下面是一个示例：

import pandas as pd

# 创建一个DataFrame
data = {'A': ['1', '2', '3', '4', '5'],
        'B': ['5', '4', '3', '2', '1'],
        'C': ['a', 'b', 'c', 'd', 'e']}
df = pd.DataFrame(data)

# 将'A'列的数据转化为整数类型
df['A'] = pd.to_numeric(df['A'])

# 将'B'列的数据转化为浮点数类型
df['B'] = pd.to_numeric(df['B'])

# 将'C'列的数据转化为数字，如果遇到无法转化的情况，用errors='coerce'参数将该值转化为NaN
df['C'] = pd.to_numeric(df['C'], errors='coerce')

print(df.dtypes)
print(df)

运行以上代码，我们得到的输出如下：

A      int64
B    float64
C    float64
dtype: object

   A    B    C
0  1  5.0  NaN
1  2  4.0  NaN
2  3  3.0  NaN
3  4  2.0  NaN
4  5  1.0  NaN

可以看到，通过使用to_numeric()函数，我们成功将字符串转化为数字，同时处理了无法转化的情况。

3. 使用apply()函数

除了以上两种方法，还可以使用apply()函数配合lambda表达式将字符串转化为数字。下面是一个示例：

import pandas as pd

# 创建一个DataFrame
data = {'A': ['1', '2', '3', '4', '5'],
        'B': ['5', '4', '3', '2', '1']}
df = pd.DataFrame(data)

# 将'A'列的数据转化为整数类型
df['A'] = df['A'].apply(lambda x: int(x))

# 将'B'列的数据转化为浮点数类型
df['B'] = df['B'].apply(lambda x: float(x))

print(df.dtypes)
print(df)

运行以上代码，我们得到的输出如下：

A      int64
B    float64
dtype: object

   A    B
0  1  5.0
1  2  4.0
2  3  3.0
3  4  2.0
4  5  1.0

通过使用apply()函数配合lambda表达式，我们同样可以将字符串转化为数字。

结论

在数据处理过程中，将字符串转化为数字是一项非常常见的操作。本文介绍了使用pandas库的三种方法将字符串转化为数字：astype()方法、to_numeric()函数和apply()函数。以上三种方法均可以实现将字符串转化为数字的功能，具体选择哪种方法可以根据实际情况来决定。

pandas把字符串转化为数字