检查数组是否是另一个数组python的元素[重复]

Question

我有一个 3d 点的 python 数组，例如 [p1,p2,...,pn] 其中 p1 = [x1,y1,zi] 我想检查天气特定点 p_i 是其中的成员，什么这是正确的方法吗？

这是我尝试过的代码

import numpy
my_list = []
for x in range(0,10):
    for y in range(0,10):
        for z  in range(0,5):
            p1 = numpy.array([x,y,z])
            my_list.append(p1)

check_list = numpy.array([[1,2,3],[20,0,20],[5,5,5]])
for p in check_list :
    if p not in my_list:
        print (p)

但是我得到了错误

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

显然这种方法适用于字符串和数字，但不适用于数组。这样做的正确方法是什么？

Answer 1

import numpy

my_list = []
for x in range(0, 10):
    for y in range(0, 10):
        for z in range(0, 5):
            p1 = numpy.array([x, y, z])
            my_list.append(p1)

check_list = numpy.array([[1, 2, 3], [20, 0, 20], [5, 5, 5]])
for p in check_list:
    if not numpy.any(numpy.all(p == my_list, axis=1)):
        print(p)

Answer 2

尝试沿轴=1 使用all 并检查结果中的any 是否为True：

for p in check_list:
    if not (p==my_list).all(1).any():
        print(p)

输出：

[20  0 20]
[5 5 5]

Answer 3

试试下面的

from dataclasses import dataclass
from typing import List


@dataclass
class Point:
    x: int
    y: int
    z: int


points: List[Point] = []
for x in range(0, 10):
    for y in range(0, 10):
        for z in range(0, 5):
            points.append(Point(x, y, z))
print(Point(1, 2, 3) in points)
print(Point(1, 2, 3888) in points)