如何从列表中选择值？

Question

我想选择标签为“1”的值，然后从所选值中随机选择1000条注释。你知道吗

这是创建上述输出的代码的文本版本：

with open('movie_ratings.txt', 'r') as f:
    lines = f.read().splitlines()
lines = lines[1:]

sentences = [line.split('\t') for line in lines]
sentences

Answer 1

仅选择最后一个元素为“1”的句子：

sentences = [line.split('\t') for line in lines if line.split('\t')[-1] == '1']

对于随机选择，这可以帮助您前进：Python random.choice() function to select random item from a List and Set

Answer 2

线路首次加载

lines = open('movie_ratings.txt').read().splitlines()[1:]
sentences = [line.split('\t') for line in lines]

现在我们将注释保留在最后一个值为“1”的位置

comments_to_keep = [
    comment for rating_id, comment, flag in sentences
    if flag == '1'
]

现在我们从这些评论中抽取一个样本

import random
sample = random.sample(comments_to_keep, 1000)