我会给你两个答案,一个只返回最大值,一个返回 CSV 中包含最大值的行。
import csv
import operator as op
import requests
symbol = "mtgoxUSD"
url = 'http://api.bitcoincharts.com/v1/trades.csv?symbol={}'.format(symbol)
csv_file = "trades_{}.csv".format(symbol)
data = requests.get(url)
with open(csv_file, "w") as f:
f.write(data.text)
with open(csv_file) as f:
next(f) # discard first row from file -- see notes
max_value = max(row[0] for row in csv.reader(f))
with open(csv_file) as f:
next(f) # discard first row from file -- see notes
max_row = max(csv.reader(f), key=op.itemgetter(0))
笔记:
max()
可以直接使用一个迭代器,并csv.reader()
给我们一个迭代器,所以我们可以把它传入。我假设你可能需要扔掉一个标题行,所以我展示了如何做到这一点。如果您有多个标题行要丢弃,您可能希望islice()
从itertools
模块中使用。
在第一个中,我们使用“生成器表达式”从每一行中选择一个值,并找到最大值。这与“列表推导”非常相似,但它不会构建整个列表,它只是让我们迭代结果值。然后max()
消耗 iterable,我们得到最大值。
max()
可以使用key=
指定“关键功能”的参数。它将使用 key 函数获取一个值并使用该值来计算最大值......但返回的值max()
将是未修改的原始值(在这种情况下,是来自 CSV 的行值)。在这种情况下,键函数是为您制造的operator.itemgetter()
......您传入您想要的列,并operator.itemgetter()
为您构建一个获取该列的函数。
结果函数等价于:
def get_col_0(row):
return row[0]
max_row = max(csv.reader(f), key=get_col_0)
或者,人们将lambda
为此使用:
max_row = max(csv.reader(f), key=lambda row: row[0])
但我认为operator.itemgetter()
阅读起来很方便。而且速度很快。
- 我展示了将数据保存在文件中,然后再次从文件中提取。如果您想浏览数据而不将其保存在任何地方,您只需要逐行遍历它。
也许是这样的:
text = data.text
rows = [line.split(',') for line in text.split("\n") if line]
rows.pop(0) # get rid of first row from data
max_value = max(row[0] for row in rows)
max_row = max(rows, key=op.itemgetter(0))
- 我不知道你想要哪一列...列“A”可能是第 0 列,所以我在上面使用了 0。根据需要替换列号。