【问题标题】:finding elements of array in another array在另一个数组中查找数组的元素
【发布时间】:2018-01-30 16:25:20
【问题描述】:

我有这段代码,我试图找出 t 数组的元素是否出现在 c1 数组中

预期结果应该是 t1 , t3 , t6

import numpy as np

t1=np.array([1,1,0,0])
t2=np.array([0,0,1,1])
t3=np.array([1,0,1,0])
t4=np.array([0,1,0,1])
t5=np.array([1,0,0,1])
t6=np.array([0,1,1,0])
two_matrix=np.array([t1,t2,t3,t4,t5,t6])
c1=[1,1,1,0]
for item in two_matrix:
    if item is part of c1 : # some code
       print(item)

【问题讨论】:

  • @Piinthesky 是的,你是对的 .edited

标签: python python-3.x numpy


【解决方案1】:

这可能是你想要的,但你需要澄清一下。

import numpy as np

t = {}
t[1] = np.array([1,1,0,0])
t[2] = np.array([0,0,1,1])
t[3] = np.array([1,0,1,0])
t[4] = np.array([0,1,0,1])
t[5] = np.array([1,0,0,1])
t[6] = np.array([0,1,1,0])

c = np.array([1,1,1,0])

for k, v in t.items():
    if all(c[i] == v[i] for i in range(len(v)) if v[i] == 1):
        print(v)

# [1 1 0 0]
# [1 0 1 0]
# [0 1 1 0]

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2016-09-24
    • 2011-02-27
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多