从python中的csv文件中获取唯一值

时间:2015-04-23 09:41:50

标签: python list api twitter set

我在Python中有以下用于twitter api的代码,我如何迭代以从CSV文件中获取值。

get_tweet = set()
gettweet_list = []
with open(TWEET_FILE) as in_file:
    for line in in_file:
        gettweet_list.append(str(line))

get_tweet.update(set(gettweet_list))
del gettweet_list
if len(get_tweet) > 140: 
 get_tweet = get_tweet[:140]
 return get_tweet

1 个答案:

答案 0 :(得分:0)

如果您只想获得第一行的第一行:

with open(TWEET_FILE) as in_file:
    data = set(next(in_file).split())

您也无法索引集合,因此get_tweet[:140]永远不会有效。

如果你想要一组所有行:

import csv
from itertools import chain
with open(TWEET_FILE) as in_file:
     data = set(chain.from_iterable(csv.reader(in_file)))

集合也没有顺序,所以即使把它变回列表并且切片也不会给你第一行。