我正在尝试使用pandas读取csv文件,而阅读时我只想读取几行和几列,因此我可以在pd.read_csv内传递所有参数,例如跳过行,行。但是我不想这样做,我想创建一个单独的函数并将该函数传递给pd.read_csv,如下所示。谁能告诉我如何实现这一目标。
import pandas as pd
file = r'C:\Users\Desktop\raw_data\Parameters.csv'
def readheader():
parse_config
{
'nrows': 200,
'header': [1,4],
'skip_blank_lines': True
}
df = pd.read_csv(file, **parse_config)
print(df)
df.head()
您可以这样做:
def someFunction(filepath_or_buffer, skip_blank_lines):
parse_config = {
'filepath_or_buffer': filepath_or_buffer,
'skip_blank_lines': skip_blank_lines
}
df = pd.read_csv(**parse_config)
someFunction(filepath_or_buffer='/some_path', skip_blank_lines=False)
您的parse_config对象的键必须有效parameters