目录
NumPy基础及取值操作
第1关:ndarray对象
import numpy as np
def print_ndarray(input_data):
'''
实例化ndarray对象并打印
:param input_data: 测试用例,类型为字典类型
:return: None
'''
#********* Begin *********#
a=np.array(input_data.get('data'))
print(a)
#********* End *********#
输入数据为字典类型,使用函数get()获取键值data并输出就可以了。
第2关:形状操作
import numpy as np
def reshape_ndarray(input_data):
'''
将ipnut_data转换成ndarray后将其变形成一位数组并打印
:param input_data: 测试用例,类型为list
:return: None
'''
#********* Begin *********#
a=np.array(input_data)
a=a.reshape(1,-1)
print(a[0])
#********* End *********#
注意input_data是函数的参数,可以直接调用,不必重新输入。
使用reshape函数将行的维度变为1,即原来的数组由二维数组变成1*x的形式,但是原来的二维数组的数组形式没有变,所以输出第一行。
第3关:基础操作
import numpy as np
def get_answer(input_data):
'''
将input_data转换成ndarray后统计每一行中最大值的位置并打印
:param input_data: 测试用例,类型为list
:return: None
'''
#********* Begin *********#
info = np.array(input_data)
print(info.argmax(axis=1))
#********* End *********#