使用 Numpy 过滤数组字符串

Question

我如何编写一个仅过滤出以“USD”结尾的数组字符串的 numpy 函数。如果没有 for 循环，我将如何执行此过滤器。

import numpy as np
Array= ['BTCUSD', 'ETHUSD', 'David', 'georGe', 'XRPUSD', 'USDAUD', 'ETHUSDC' ]

预期输出

['BTCUSD', 'ETHUSD',  'XRPUSD']

Answer 1

使用 numpy char.endswith

import numpy as np

a = np.array(['BTCUSD', 'ETHUSD', 'David', 'georGe', 'XRPUSD', 'USDAUD', 'ETHUSDC'])
print(a[np.char.endswith(a, 'USD')])

输出：

['BTCUSD' 'ETHUSD' 'XRPUSD']

---

对于list 而不是np.ndarray 的返回类型，可以使用推导式：

import numpy as np

lst = np.array(['BTCUSD', 'ETHUSD', 'David', 'georGe', 'XRPUSD', 'USDAUD', 'ETHUSDC'])
print([elem for elem in lst if elem.endswith('USD')])

输出：

['BTCUSD', 'ETHUSD', 'XRPUSD']

*理解方法可用于 Python 列表以及 np 数组。