從python中的csv文件獲取唯一值

Question

我在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

Answer 1

如果您只想要第一行，如何獲得第一行：

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)))

集合也沒有順序，因此即使將其返回到列表中，切片也不會給您第一行。