如何计算二维数组中包含某个字符串的元素个数?

2024-04-20 03:47:03 发布

您现在位置:Python中文网/ 问答频道 /正文

我有这样一个二维数组:

[[aaa-1-2,
  aaa-3-4,
  bbb-y-t,
  aaa-5-t],
[bbb-1-2,
  dfs-3-4,
  bbb-a-9,
  yui-5-t]]

我想写一个函数或者别的什么来得到包含aaa的元素的和(在[aaa-1-2, aaa-3-4, bbb-y-t, aaa-5-t]),和bbb的和(在[bbb-1-2, dfs-3-4, bbb-a-9, yui-5-t])。你知道吗

在这个例子中,和应该是3+2=5,我只知道如何计算两个数组之间完全相同的元素的和,但不知道如何解决这个问题,有人能帮我吗?谢谢。你知道吗


Tags: 函数元素数组例子bbbyuiaaadfs
1条回答
网友
1楼 · 发布于 2024-04-20 03:47:03

对于有n个前缀对应于n个子列表的一般解决方案,可以使用enumerate找出您所在的子列表,并查找要使用的适当前缀,然后对测试求和:

l = [["aaa-1-2", "aaa-3-4","bbb-y-t","aaa-5-t"],
     ["bbb-1-2", "dfs-3-4", "bbb-a-9","yui-5-t"]]

prefix = ['aaa', 'bbb']    
sum(prefix[i] in s for i, sub in enumerate(l) for s in sub)

# 5

如果你有更多的子列表和更多的刺你可以添加到前缀列表。你知道吗

相关问题 更多 >