如何将提取df['genres']中id的嵌套循环改写为Python列表推导式

改写后的列表推导式写法 #

你给出的嵌套循环可以直接按照循环顺序对应改写为两层嵌套的列表推导式，代码如下：

lista = [[genres['id'] for genres in dict_genres] for dict_genres in df['genres']]

逻辑说明 #

• 推导式的循环顺序和原有代码的循环顺序完全一致：最外层的for dict_genres in df['genres']对应原代码的第一层循环，内层的[genres['id'] for genres in dict_genres]对应原代码里内层遍历genres取id、生成suport列表的逻辑，不需要额外定义中间变量。
• 改写后和原有代码的执行效果完全相同，代码更简洁。

内容的提问来源于stack exchange，提问作者andrdpedro