将inputfile读入python中的csv

Question

我想从STDIN读取csv文件并对其进行操作。

下面是读取csv文件并执行所需操作的代码。这个很好用。但我想从STDIN那里得到信息。

import csv
with open('hospital_data.csv', 'rb') as csvfile:
    myDict = {}

    csvreader = csv.reader(csvfile, delimiter=',')
    for row in csvreader:
        if row[6] not in myDict.keys():
            #print 'Zipcode: ' + row[6] + ' Hospital code: ' + row[1]
            myDict[row[6]] = 1
        elif row[6] in myDict.keys():
            #print 'value in row '+ str(myDict[row[6]])
            myDict[row[6]] += 1

Python中是否有一种方法可以将STDIN中的文件作为csv文件读取？

Answer 1

csv.reader将获取任何产生行的内容，因此您可以使用这个答案中所示的任何方法从stdin：How do you read from stdin in Python?获取行。

由于fileinput的灵活性，我本人也喜欢它。例：

import csv
import fileinput

myDict = {}
csvreader = csv.reader(fileinput.input(mode='rb'), delimiter=',')

但这也适用于：

import csv
import sys

myDict = {}
csvreader = csv.reader(sys.stdin, delimiter=',')

如果这样做，您将希望使用-u命令行参数来生成流二进制，如果这在您的平台上产生了影响：https://docs.python.org/2/using/cmdline.html#cmdoption-u

在这两种情况下，您都需要使用control来标记输入的结尾。

注意，检查键是否在dict中的正确方法是if row[6] in myDict，而不是检查keys。事实上，如果您只希望在键不存在时使用默认值，请使用get

for row in csvreader:
    myDict[row[6]] = myDict.get(row[6], 0) + 1

或者看看collections.Counter，因为你用的是2.7：

myDict = collections.Counter(row[6] for row in csvreader)

Answer 2

使用sys.stdin，它是一个类似文件的对象.

import sys
import csv

data = sys.stdin.readlines()
csvreader = csv.reader(data, delimiter=',')
for row in csvreader:
    print row