在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})