208

有时将行或列向量“克隆”到矩阵中很有用。通过克隆，我的意思是转换行向量，例如

[1, 2, 3]

成矩阵

[[1, 2, 3],
 [1, 2, 3],
 [1, 2, 3]]

或列向量，例如

[[1],
 [2],
 [3]]

进入

[[1, 1, 1]
 [2, 2, 2]
 [3, 3, 3]]

在 MATLAB 或 octave 中，这很容易完成：

 x = [1, 2, 3]
 a = ones(3, 1) * x
 a =

    1   2   3
    1   2   3
    1   2   3
    
 b = (x') * ones(1, 3)
 b =

    1   1   1
    2   2   2
    3   3   3

我想在 numpy 中重复这个，但没有成功

In [14]: x = array([1, 2, 3])
In [14]: ones((3, 1)) * x
Out[14]:
array([[ 1.,  2.,  3.],
       [ 1.,  2.,  3.],
       [ 1.,  2.,  3.]])
# so far so good
In [16]: x.transpose() * ones((1, 3))
Out[16]: array([[ 1.,  2.,  3.]])
# DAMN
# I end up with 
In [17]: (ones((3, 1)) * x).transpose()
Out[17]:
array([[ 1.,  1.,  1.],
       [ 2.,  2.,  2.],
       [ 3.,  3.,  3.]])

为什么第一种方法 ( In [16]) 不起作用？有没有办法以更优雅的方式在 python 中完成这项任务？

4

12 回答 12

390

使用numpy.tile：

>>> tile(array([1,2,3]), (3, 1))
array([[1, 2, 3],
       [1, 2, 3],
       [1, 2, 3]])

或用于重复列：

>>> tile(array([[1,2,3]]).transpose(), (1, 3))
array([[1, 1, 1],
       [2, 2, 2],
       [3, 3, 3]])
于 2009-10-17T17:33:21.667 回答
104

这是一种优雅的 Pythonic 方式：

>>> array([[1,2,3],]*3)
array([[1, 2, 3],
       [1, 2, 3],
       [1, 2, 3]])

>>> array([[1,2,3],]*3).transpose()
array([[1, 1, 1],
       [2, 2, 2],
       [3, 3, 3]])

问题[16]似乎是转置对数组没有影响。您可能想要一个矩阵：

>>> x = array([1,2,3])
>>> x
array([1, 2, 3])
>>> x.transpose()
array([1, 2, 3])
>>> matrix([1,2,3])
matrix([[1, 2, 3]])
>>> matrix([1,2,3]).transpose()
matrix([[1],
        [2],
        [3]])
于 2009-10-11T07:59:23.153 回答
57

首先请注意，使用 numpy 的广播操作，通常不需要复制行和列。有关说明，请参见this和this。

但要做到这一点，repeat和newaxis可能是最好的方法

In [12]: x = array([1,2,3])

In [13]: repeat(x[:,newaxis], 3, 1)
Out[13]: 
array([[1, 1, 1],
       [2, 2, 2],
       [3, 3, 3]])

In [14]: repeat(x[newaxis,:], 3, 0)
Out[14]: 
array([[1, 2, 3],
       [1, 2, 3],
       [1, 2, 3]])

此示例适用于行向量，但希望将其应用于列向量是显而易见的。repeat 似乎拼写得很好，但你也可以通过乘法来做到这一点，就像你的例子一样

In [15]: x = array([[1, 2, 3]])  # note the double brackets

In [16]: (ones((3,1))*x).transpose()
Out[16]: 
array([[ 1.,  1.,  1.],
       [ 2.,  2.,  2.],
       [ 3.,  3.,  3.]])
于 2009-10-11T22:56:20.577 回答
16

让：

>>> n = 1000
>>> x = np.arange(n)
>>> reps = 10000

零成本分配

视图不占用任何额外的内存。因此，这些声明是即时的：

# New axis
x[np.newaxis, ...]

# Broadcast to specific shape
np.broadcast_to(x, (reps, n))

强制分配

如果要强制内容驻留在内存中：

>>> %timeit np.array(np.broadcast_to(x, (reps, n)))
10.2 ms ± 62.3 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

>>> %timeit np.repeat(x[np.newaxis, :], reps, axis=0)
9.88 ms ± 52.4 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

>>> %timeit np.tile(x, (reps, 1))
9.97 ms ± 77.3 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

这三种方法的速度大致相同。

计算

>>> a = np.arange(reps * n).reshape(reps, n)
>>> x_tiled = np.tile(x, (reps, 1))

>>> %timeit np.broadcast_to(x, (reps, n)) * a
17.1 ms ± 284 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

>>> %timeit x[np.newaxis, :] * a
17.5 ms ± 300 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

>>> %timeit x_tiled * a
17.6 ms ± 240 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

这三种方法的速度大致相同。

结论

如果您想在计算之前进行复制，请考虑使用“零成本分配”方法之一。您不会遭受“强制分配”的性能损失。

于 2017-07-15T12:51:54.697 回答
9

我认为在 numpy 中使用广播是最好的，而且速度更快

我做了如下比较

import numpy as np
b = np.random.randn(1000)
In [105]: %timeit c = np.tile(b[:, newaxis], (1,100))
1000 loops, best of 3: 354 µs per loop

In [106]: %timeit c = np.repeat(b[:, newaxis], 100, axis=1)
1000 loops, best of 3: 347 µs per loop

In [107]: %timeit c = np.array([b,]*100).transpose()
100 loops, best of 3: 5.56 ms per loop

使用广播大约快 15 倍

于 2014-01-15T19:10:59.507 回答
5

一种干净的解决方案是使用 NumPy 的外积函数和一个向量：

np.outer(np.ones(n), x)

给出n重复的行。切换参数顺序以获取重复列。要获得相同数量的行和列，您可能会这样做

np.outer(np.ones_like(x), x)
于 2019-01-17T06:18:00.217 回答
3

您可以使用

np.tile(x,3).reshape((4,3))

tile 将生成向量的代表

和 reshape 会给它你想要的形状

于 2016-12-07T12:20:40.597 回答
2

回到原来的问题

在 MATLAB 或 octave 中，这很容易完成：

x = [1, 2, 3]

a = 个(3, 1) * x ...

在 numpy 中它几乎是一样的（也很容易记住）：

x = [1, 2, 3]
a = np.tile(x, (3, 1))
于 2021-03-12T23:01:58.527 回答
1

如果您有一个 pandas 数据框并且想要保留 dtypes，甚至是分类，这是一种快速的方法：

import numpy as np
import pandas as pd
df = pd.DataFrame({1: [1, 2, 3], 2: [4, 5, 6]})
number_repeats = 50
new_df = df.reindex(np.tile(df.index, number_repeats))
于 2019-11-10T03:47:06.090 回答
1

另一种解决方案

>> x = np.array([1,2,3])
>> y = x[None, :] * np.ones((3,))[:, None]
>> y
array([[ 1.,  2.,  3.],
       [ 1.,  2.,  3.],
       [ 1.,  2.,  3.]])

为什么？当然，重复和平铺是正确的方法。但是无索引是一个强大的工具，它多次让我快速矢量化一个操作（尽管它很快就会非常消耗内存！）。

我自己的代码中的一个例子：

# trajectory is a sequence of xy coordinates [n_points, 2]
# xy_obstacles is a list of obstacles' xy coordinates [n_obstacles, 2]
# to compute dx, dy distance between every obstacle and every pose in the trajectory
deltas = trajectory[:, None, :2] - xy_obstacles[None, :, :2]
# we can easily convert x-y distance to a norm
distances = np.linalg.norm(deltas, axis=-1)
# distances is now [timesteps, obstacles]. Now we can for example find the closest obstacle at every point in the trajectory by doing
closest_obstacles = np.argmin(distances, axis=1)
# we could also find how safe the trajectory is, by finding the smallest distance over the entire trajectory
danger = np.min(distances)
于 2021-06-18T12:00:18.147 回答
0

为了回答实际问题，现在已经发布了近十种解决解决方案的方法：x.transpose反转x. 一个有趣的副作用是，如果x.ndim == 1，转置什么也不做。

这对于来自 MATLAB 的人来说尤其令人困惑，因为所有数组都隐含地至少有两个维度。转置一维 numpy 数组的正确方法不是x.transpose()or x.T，而是

x[:, None]

或者

x.reshape(-1, 1)

从这里，您可以乘以一个矩阵，或使用任何其他建议的方法，只要您尊重 MATLAB 和 numpy 之间的（细微）差异。

于 2021-01-28T21:53:54.833 回答
-1 
import numpy as np
x=np.array([1,2,3])
y=np.multiply(np.ones((len(x),len(x))),x).T
print(y)

产量：

[[ 1.  1.  1.]
 [ 2.  2.  2.]
 [ 3.  3.  3.]]
于 2018-03-17T09:07:59.427 回答