Question

我写了一个Python脚本来分析CSV文件。我想实现一个文件选择器，所以当我执行脚本时，它应该打开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的其他信息。

读取csv文件作为python中的变量

1 个答案: