过滤多数组字符串

Question

我想编写一个 numpy 函数，将 a 数组过滤为仅以 USD 和 USDC 结尾的数组。下面的函数仅适用于一个过滤器，但不适用于两个过滤器 'USD', 'USDC'。代码已从问题中获取：issue

import numpy as np

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

Answer 1

试试这个：

import numpy as np

a = np.array(['BTCUSD', 'ETHUSD', 'David', 'georGe', 'XRPUSD', 'USDAUD', 'ETHUSDC'])
b = np.array([i for i in a if i.endswith('USDC') or i.endswith('USD')])
print(b)