一个简单的例子

n = 5
a = np.array([i for i in range(n)])
print(a)
>>> output: [0 1 2 3 4]

循环中加入if

X = np.array([1, 2, 3, 4])
Y = [X[i] for i in range(len(X)) if X[i] % 2 == 0]
print(Y)
>>> output: [2, 4]
Logo

助力广东及东莞地区开发者,代码托管、在线学习与竞赛、技术交流与分享、资源共享、职业发展,成为松山湖开发者首选的工作与学习平台

更多推荐