Python numpy.add函数的具体使用

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

result = np.add(arr1, arr2)
print(result)
# 输出: [5 7 9]

matrix1 = np.array([[1, 2], [3, 4]])
matrix2 = np.array([[5, 6], [7, 8]])

result_matrix = np.add(matrix1, matrix2)
print(result_matrix)
# 输出:
# [[ 6  8]
#  [10 12]]

constant = 10
result_with_constant = np.add(arr1, constant)
print(result_with_constant)
# 输出: [11 12 13]

matrix = np.array([[1, 2], [3, 4]])
row_vector = np.array([5, 6])

result_broadcast = np.add(matrix, row_vector)
print(result_broadcast)
# 输出:
# [[ 6  8]
#  [ 8 10]]

def custom_function(x):
    return x**2 + 1

arr = np.array([1, 2, 3])
result_custom_function = np.add(arr, 5)
result = custom_function(result_custom_function)
print(result)
# 输出: [ 7 10 16]

arr = np.array([np.inf, np.nan, 5, 10])
constant = 5

result_no_casting = np.add(arr, constant)
print(result_no_casting)
# 输出: [inf nan 10. 15.]

result_with_casting = np.add(arr, constant, casting='unsafe')
print(result_with_casting)
# 输出: [inf nan 10. 15.]

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
output = np.zeros(3)

np.add(arr1, arr2, out=output)
print(output)
# 输出: [5. 7. 9.]

add_custom = np.frompyfunc(lambda x, y: x + y, 2, 1)

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

result_custom = add_custom(arr1, arr2)
print(result_custom)
# 输出: [5 7 9]

import timeit

arr = np.random.rand(1000000)
constant = 5

# 使用numpy.add
time_numpy = timeit.timeit(lambda: np.add(arr, constant), number=100)

# 使用纯Python循环
time_python = timeit.timeit(lambda: [x + constant for x in arr], number=100)

print("Time with numpy.add:", time_numpy)
print("Time with Python loop:", time_python)