unique_rarity = list(set(rarity)) data_rarity = [] for r in unique_rarity: data_rarity = [x[2] for x in data if x[4] == r]优化这段代码

可以使用列表推导式和字典来优化这段代码```
data_rarity =r: [x[] for x in data if x[4] == r] for r in set(rarity)}

这样就一行代码快速地创建字典，其中键rarity` 列表的唯一元素，对应的值是 `data` 列表中与该元素匹配的所有 `x[2]`。这样做的好处是避免了多次遍历 `data` 列表，提高了代码效率。