np.repeat用法及参数
时间: 2023-11-20 12:06:04 浏览: 94
python多项式拟合之np.polyfit 和 np.polyld详解
np.repeat函数用于将一个数组中的元素重复指定的次数,返回一个新的数组。
参数:
- a: 数组或列表,需要重复的元素。
- repeats: int或数组,指定每个元素需要重复的次数。如果是一个整数,则所有元素都将重复相同的次数;如果是一个数组,则指定每个元素需要重复的不同次数。
- axis: 指定在哪个轴上进行重复操作。默认为None,表示将数组展平后进行重复操作。
示例:
```python
import numpy as np
a = np.array([1, 2, 3])
b = np.repeat(a, 3)
print(b) # [1 1 1 2 2 2 3 3 3]
c = np.repeat(a, [1, 2, 3])
print(c) # [1 2 2 3 3 3]
d = np.array([[1, 2], [3, 4]])
e = np.repeat(d, 2, axis=0)
print(e)
# [[1 2]
# [1 2]
# [3 4]
# [3 4]]
f = np.repeat(d, 2, axis=1)
print(f)
# [[1 1 2 2]
# [3 3 4 4]]
```
阅读全文