python - 如何将此 python 行代码转换为列表理解？

Question

我被要求将这些行转换为列表理解

tmaxima = []
ag_maxima = []
for i in sorted_indices:
tmaxima.append(t[i])
ag_maxima.append(a_g[i])

标签： pythonlistfor-looplist-comprehensionspyder

Answer 1

假设sorted_indices仅包含整数：

tmaxima = [t[i] for i in sorted_indices]
ag_maxima = [a_g[i] for i in sorted_indices]