python - 如何在函数之间解析 pandas 数据帧

Question

我想将 csv 文件中的数据读入 pandas 数据帧。然后我想对这个数据框进行一些操作。我想在不同的函数中执行此操作(最好在单独的文件中)。

import pandas as pd

def read_text(file):
    df = pd.read_csv(file,skipinitialspace=True, sep=";", encoding = "ISO-8859-1")
    return [df]


file = "/path/file.txt"
content = pd.DataFrame()
content = read_text(file)

现在，文件的读取工作正常。但“内容”似乎不再是数据框了。至少，如果我尝试类似的事情print(content.value) 好像没有这个选项。我后来想做的是:

1. 将数据帧发送到函数以删除重复项并返回数据帧
2. 获取这个新数据帧并删除某些条目，然后再次返回该数据帧
3. 使用数据框做更多事情

理想情况下，这些函数应位于单独的文件中。但我稍后会处理这个问题。现在，如果我可以来回解析这些数据帧，那将会有很大的帮助。

Answer 1

您将返回[df]，因此这是单个数据帧的列表。您应该按如下方式修改代码:

import pandas as pd

def read_text(file):
    df = pd.read_csv(file,skipinitialspace=True, sep=";", encoding = "ISO-8859-1")
    return df


file = "/path/file.txt"
content = read_text(file)