在pandas中读取csv上的数据

Question

我正在将.csv文件读入pandas数据帧。 .csv文件包含多个列。 列'A'包含字符串'20 -989-98766'。 是否可以在加载文件时从字符串中读取最后5个字符'98766'？

df = pd.read_csv("test_data2.csv", column={'A':read the last 5 characters})

输出：

A
98766
95476
.....

Answer 1

您可以定义一个func并将其作为arg传递给read_csv converters参数：

In [57]:
import io
import pandas as pd
def func(x):
    return x[-5:]
t="""column
'20-989-98766"""
df = pd.read_csv(io.StringIO(t), converters={'column': func})
df

​
Out[57]:
  column
0  98766

所以在这里我定义一个func并以dict的形式将其传递给converters ，并以列名作为键，这将在csv中的每一行调用func

所以在你的情况下，以下应该工作：

df = pd.read_csv("test_data2.csv", converters={'A':func})