numpy Python如何计算包含字符串的嵌套列表

jecbmhm3  于 2023-03-30  发布在  Python
关注(0)|答案(1)|浏览(94)

我有一个嵌套的列表列表。我想知道主列表和子列表中的总项目。然后相应地我想为每个子列表或项目选择一个值。
我的代码:

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]
1aaf6o9v

1aaf6o9v1#

您可以检查每个元素是否是列表,并以不同的方式处理它们。

print([len(x) if isinstance(x, list) else 1 for x in y]) 
# [1, 2, 1]
plot_alpha = [alpha if isinstance(x, list) else 1 for x in y]
# [1, 0.5, 1]

相关问题