当我们需要对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数据框架中把字符串转换成整数。