从用户输入中获取信息，然后根据他们的输入为一个.txt文件和一个.csv文件命名。

Question

我正在尝试制作一个程序，用户需要输入关键字，然后我的程序将使用Twitter API返回包含该关键字的推文。我还希望它能够根据用户输入的关键字命名文件。假设他们输入了“iphone”，我想让它生成一个iphone.txt和iphone.csv文件。以下是我目前的代码，但似乎无法正常工作。

顺便说一下，“newFile”行应该是一个.txt文件，“with open(query, 'rb') as input_file”也应该是一个.txt文件。另一个是.csv文件。

try:
    query = sys.argv[1]
except IndexError:
    query = raw_input("Choose a keyword to find the last 100 tweets about: ")

newFile = open(query, 'w').write(txt.encode('utf8'))


with open(query, 'rb') as input_file:
    reader = csv.reader(input_file, delimiter='\n', quoting = csv.QUOTE_NONE)

    with open(query, 'wb') as output_file:
        writer = csv.writer(output_file)

        for row in reader:
            writer.writerow(row)

- Neemaximo

Answer 1

就快完成了，但是你的语句应该是：

with open("%s.csv" %query, 'rb') as input_file

以及

with open("%s.txt" %query, 'wb') as output_file

query 只包含值，即文件名，在打开文件之前需要添加扩展名。

- avasal