numpy.ix_()函数

调用方法:

numpy.ix_(*args)

各个参数意义

args:1-D sequences,应该为整数或者布尔数
有返回值,返回一个ndarray元组,n个array返回的元组就是n维的

代码如下:

# -*- coding:utf-8 -*-
"""
author: 15025
time: 2020/11/19 12:21
software: PyCharm

Description:
"""
import numpy as np


class Numpy:
    @staticmethod
    def mainProgram():
        array = np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])
        index = np.ix_([0, 1], [2, 4])
        print("array为: ")
        print(array[index])


if __name__ == "__main__":
    n = Numpy()
    n.mainProgram()
"""
array为: 
[[ 3  5]
 [ 8 10]]
"""

猜你喜欢

转载自blog.csdn.net/u011699626/article/details/110009835