最近有一个需求,在不知道numpy维度的情况下,对最后一个维度的数据进行操作,假设需要对最后一个维度的第一行数据加1:
def add(narr):
# 未知narr维度, '::'指代前面所有的维度数据
arr[::, 0] = arr[::, 0] + 1
return arr
最近有一个需求,在不知道numpy维度的情况下,对最后一个维度的数据进行操作,假设需要对最后一个维度的第一行数据加1:
def add(narr):
# 未知narr维度, '::'指代前面所有的维度数据
arr[::, 0] = arr[::, 0] + 1
return arr