我有一个嵌套的列表列表。我想知道主列表和子列表中的总项目。然后相应地我想为每个子列表或项目选择一个值。
我的代码:
y = ['ab',['bc','cd'],'ef']
print([len(y[i]) for i in range(len(y))])
alpha=0.5
plot_alpha = [alpha for i in range(len(y)) if len(y[i])>1 else 0.5]
print(plot_alpha)
我现在的回答是:
[2, 2, 2]
[0.5, 0.5, 0.5]
预期答案:
[1, 2, 1]
[1, 0.5, 1]
1条答案
按热度按时间1aaf6o9v1#
您可以检查每个元素是否是列表,并以不同的方式处理它们。