如何在Pandas数据框架中把字符串转换成整数

  • Post category:Python

当我们需要对Pandas数据框架中的字符串数据进行数值计算时,往往需要将其转换为整数类型。下面将介绍如何在Pandas中把字符串转换为整数。

1.读取数据

首先,我们需要读取数据并创建Pandas数据框架。本文以读取一个名为“data.csv”的数据文件为例。

import pandas as pd

#读取csv文件
df = pd.read_csv('data.csv')

#输出数据框前几行
print(df.head())

2.查看数据信息

在数据处理之前,我们需要先了解数据的信息,以便进行下一步的操作。 这时我们可以使用info()函数,查看整个数据框架的信息及数据类型。

df.info()

从信息中我们可以看到,部分列的数据类型为object,即字符串类型。

3.把字符串转换成整数

在Pandas中,我们可以使用astype()函数将数据类型转换为整数类型。下面是一个具体的例子:

#查看数据类型
print(type(df['col1'][0]))

#将对应列的字符串类型转换为int类型
df['col1_int'] = df['col1'].astype(int)

#查看转换之后的数据类型
print(type(df['col1_int'][0]))

在以上例子中,我们将一个名为“col1”的列的字符串类型转换为int类型,并新增一列“col1_int”存储转换之后的值,即将字符串类型的数据成功转换为int类型。

4.数据处理与保存

在完成所有需要的数据转换之后,我们可以将处理结果导出为csv文件。

#导出结果
df.to_csv('result.csv', index=False)

#查看导出的结果
df_result = pd.read_csv('result.csv')
print(df_result.head())

在导出结果之后,我们可以通过查看csv文件,可以发现数据框架已经将字符串型数据成功转换为整数型数据,并保存到了csv文件中。

综上所述,我们可以通过以上步骤实现在Pandas数据框架中把字符串转换成整数。