Numpy如果元素小于或等于，则传递

Question

我正在尝试对NUMPY/uplot中的值列表进行排序，但由于我是一个新的uplot用户，所以在格式方面遇到了问题

这些值位于其他一些列表中，我们将使用名称x逐个调用这些值

如果x的值大于或等于5，我想将其添加到数组中，该数组最初为空。如果数字小于5，那么我们进入下一个数字

具体来说，我需要有关如何设置“大于等于”格式的帮助

array = [] 

if x is greater than or equal to 5:
    array.append(x)
else:
    return 0

谢谢大家

Answer 1

您使用的是python列表，它不同于numpy数组。无论哪种方式，下面的代码都应该工作

X = np.random.random(size= [10])    # array containing x values

如果你想要一个numpy数组

arr = X[X >= 5]

如果你想要一份清单

arr = list(X[X >= 5])

Answer 2

使用numpy可以执行以下操作：

import numpy as np

# Initialize array
array = np.array([])
  
# Make some random values for x  
x = np.floor(np.random.rand(10)*10)
  
for i in x: # Loop through x
    if i >= 5: # If value is bigger or equal to 5
        array = np.append(array, i) # add to array

所以，“大于等于”只是>=