5

I have a W x H array A1. There is another W x M array A2, where M << H. These M points along that dimension is supposed to be put in equal-spaced cells of the for all the W dimension. I've achieved this by

hopSize = H / M
A1[:, 0 : min(A1.shape[1], hopSize*M) : hopSize] = A2

Now I want to populate the value of the M anchor points to fill all the points between those anchors along the H dimension, e.g., Anchor 1's value will be copied to every point for A1[:, Anchor1 : Anchor2].

I wonder if there is a way to achieve this without using for-loop.

1 Answer 1

5

A general approach would be to use scipy.interpolate.interp1d:

import numpy as np
from scipy.interpolate import interp1d

# generate some example data
W = 3
H = 10
M = 5

A2 = np.arange(W * M).reshape(W, M)
print(A2)
# [[ 0  1  2  3  4]
#  [ 5  6  7  8  9]
#  [10 11 12 13 14]]

# the initial column indices for A2
x = np.arange(M)

# we create a scipy.interpolate.interp1d instance
itp_A2 = interp1d(x, A2, kind='nearest')

# the output column coordinates for A1
xi = np.linspace(0, M - 1, H)

# we get the interpolated output by calling the interp1d instance with the
# output coordinates
A1 = itp_A2(xi)
print(A1)
# [[  0.   0.   1.   1.   2.   2.   3.   3.   4.   4.]
#  [  5.   5.   6.   6.   7.   7.   8.   8.   9.   9.]
#  [ 10.  10.  11.  11.  12.  12.  13.  13.  14.  14.]]

As well as nearest-neighbor interpolation you could do linear, quadratic, cubic etc.


For the special case where you are upsampling by an integer factor using nearest-neighbor interpolation, you could just use np.repeat:

# upsampling factor
fac = H / M

print(np.repeat(A2, fac, 1))
# [[ 0  0  1  1  2  2  3  3  4  4]
#  [ 5  5  6  6  7  7  8  8  9  9]
#  [10 10 11 11 12 12 13 13 14 14]]
Sign up to request clarification or add additional context in comments.

Comments

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

Start asking to get answers

Find the answer to your question by asking.

Ask question

Explore related questions

See similar questions with these tags.