如何读取以 Python pandas 中的特定子字符串开头的 CSV 文件？

Question

Say I have a CSV file whose name is like:

Pokémon_Pikachu.csv.

Is there a way to read it if I give only the first substring (Pokémon) and still read it in Pandas ?

Answer 1

import glob
import pandas as pd

for file in glob.glob("Pokémon*.csv"):
    print (file)

this will get you the csv file names that start with Pokémon and if you want to read all the csv files into one,

main_df = pd.DataFrame()
for file in glob.glob("Pokémon*.py"):
    df = pd.read_csv(file)
    if main_df.empty:
        main_df = df
    else:
        main_df = main_df.join(df, how='outer')

print main_df.head()

Answer 2

http://www.pythonforbeginners.com/code-snippets-source-code/python-os-listdir-and-endswith

you can use os.listdir() to get a list of the contents of a directory, then filter those with string.startswith(substring) or string.endswith(substring) .

That would give you the filename(s) that you could put into pd.read_csv(filename)

Answer 3

I am not sure what you mean but I suppose that you have files in directory with prefix Pokémon . The sollution is:

import pandas as pd
import os
import glob

for file in glob.glob(os.path.join(input_dir, 'Pokemon_*.csv')):
    pd.read_csv(file)