将csv文件作为python中的变量读入

Question

我编写了一个分析 CSV 文件的 Python 脚本。 我想实现一个文件选择器，所以当我执行脚本时它应该打开 Windows 资源管理器并让我选择它应该分析的文件。 到目前为止，它手动读取了 1 个文件：

import pandas as pd

df= pd.read_csv('file1.csv', usecols=['col1', 'col2'])

我想用一个变量替换 file1.csv，所以每次执行脚本时我都可以选择不同的文件。

file = open.windowsexplorer (choose file from directory)
df=pd.read_csv (' ' +  file', usecols='col1', 'col2'])

此外，我想对列做同样的事情。所以我可以输入我想选择的列名。如果在其他文件中，列的名称不同：

column1 = 在此处输入

column2 = 在此处输入（作为 Windows 警报）

Answer 1

这是一个可以满足您要求的裸代码：

import tkinter as tk
from tkinter import filedialog
from tkinter import simpledialog
import pandas as pd

root = tk.Tk()
root.withdraw()

path = filedialog.askopenfilename(parent=root, filetypes=[("CSV Files",".csv")])
col1 = simpledialog.askstring("Input", "Column 1", parent=root, initialvalue="col1")
col2 = simpledialog.askstring("Input", "Column 2", parent=root, initialvalue="col2")

df = pd.read_csv(path, usecols=[col1, col2])

您将找到有关简单对话框here 的更多信息。