如何在用戶輸入后使用 pandas 打開 a.csv 文件？

Question

我對 Python 很陌生，這將是一個非常基本的問題。 我希望用戶輸入 csv 文件的名稱，我想用 pandas 打開該文件以輕松訪問其行和列。 這是我寫的代碼：

import pandas as pd 
DATAFIN = str(raw_input("Name of your data file"))
dataset = pd.read_csv(DATAFIN)
dataset.head()

但是，我似乎犯了某種錯誤，因為這是我收到的消息（對不起，太長了）：

Traceback (most recent call last):
  File "c:\Users\File.py", line 34, in <module>
    dataset = pd.read_csv(DATAFIN)
  File "C:\Python27\lib\site-packages\pandas\io\parsers.py", line 702, in parser_f
    return _read(filepath_or_buffer, kwds)
  File "C:\Python27\lib\site-packages\pandas\io\parsers.py", line 429, in _read
    parser = TextFileReader(filepath_or_buffer, **kwds)
  File "C:\Python27\lib\site-packages\pandas\io\parsers.py", line 895, in __init__
    self._make_engine(self.engine)
  File "C:\Python27\lib\site-packages\pandas\io\parsers.py", line 1122, in _make_engine
    self._engine = CParserWrapper(self.f, **self.options)
  File "C:\Python27\lib\site-packages\pandas\io\parsers.py", line 1853, in __init__
    self._reader = parsers.TextReader(src, **kwds)
  File "pandas\_libs\parsers.pyx", line 387, in pandas._libs.parsers.TextReader.__cinit__
  File "pandas\_libs\parsers.pyx", line 705, in pandas._libs.parsers.TextReader._setup_parser_source
 does not exist: ' maindata.csv\r'csv

你知道哪個是問題嗎？ 對於格式中的任何錯誤，我深表歉意。

Answer 1

看起來您在字符串中使用了空格字符

' maindata.csv\r'

嘗試輸入您的 csv 名稱，不帶空格

所以看起來像

Name of your data filemaindata.csv

Answer 2

嘗試使用pd.read_csv(r'address of the file.csv ') 讀取您的 csv 文件

Answer 3

使用 argparse 從命令行獲取文件名。 通過python script.py --filename file.csv運行您的腳本並使用 print 查看結果

import pandas as pd 
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--filename')
args = parser.parse_args()
dataset = pd.read_csv(args.filename)
print(dataset.head())