我的python文件夹中有一个名为data的文件夹,里面装满了文本文件,每个文件都有一个州的缩写(“NY”,“CA”,“RI”等)。我需要用户输入状态名称,然后打开该文件。
我试过了:
file = ""
file = open("NY", "r").read().splitelines()
但它在目录中没有任何内容。如何从文件夹中打开单个文件?
答案 0 :(得分:2)
from os.path import join, dirname
data_dir = join(dirname(__file__), 'data')
with open(join(data_dir, raw_input('Enter state: '))) as f:
print(f.readlines())
此代码生成基本dir变量并将其与州名称连接。执行后你会看到
brunsgaard@archbook ~/t/test> python test.py
Enter state: NY
['Hello, I am the NY file\n']
此示例在以下指向性结构的上下文中运行。
.
├── data
│ └── NY
└── test.py