如果另一个数组不继续，则通过重复值来扩展数组

Question

我正在追踪一些粒子在平面上使用TrackPy插件。这将产生一个数据帧，其位置为x和y，以及相应的帧号，这里用一个简单的列表说明了这一点：

x=[80.1,80.2,80.1,80.2,80.3]
y=[40.1,40.2,40.1,40.2,40.3]
frame = [1,2,3,4,5]

但是，由于实验装置的原因，单个帧的粒子可能会丢失，从而导致：

x=[80.1,80.2,80.1,80.2,80.3]
y=[40.1,40.2,40.1,40.2,40.3]
frame = [1,2,3,4,6]

现在，我想扩展所有列表，这样“frame”将继续，如果原始数据中没有框架，则'x，y‘将重复以前的值，结果如下：

x=[80.1,80.2,80.1,80.2,80,2,80.3]
y=[40.1,40.2,40.1,40.2,40.2,40.3]
frame = [1,2,3,4,5,6]

Answer 1

您可以使用Pandas，它在内部利用NumPy数组：

import pandas as pd

df = pd.DataFrame({'x': x, 'y': y}, index=frame)

df = df.reindex(np.arange(df.index.min(), df.index.max()+1)).ffill()

结果

print(df)

      x     y
1  80.1  40.1
2  80.2  40.2
3  80.1  40.1
4  80.2  40.2
5  80.2  40.2
6  80.3  40.3

然后，可以将结果提取到列表中：

x = df['x'].tolist()
y = df['y'].tolist()
frame = df.index.tolist()

Answer 2

这里有一个基于NumPy的解决方案-

def extend_arrs(x,y,frame):
    # Convert to arrays
    frame = np.asarray(frame)
    x = np.asarray(x)
    y = np.asarray(y)

    l = frame[-1]-frame[0] + 1
    id_ar = np.zeros(l,dtype=int)
    id_ar[frame-frame[0]] = 1
    idx = id_ar.cumsum()-1
    return np.r_[frame[0]:frame[-1]+1],x[idx], y[idx]

样本运行-

In [164]: x
Out[164]: [80.1, 80.2, 80.1, 80.2, 80.3]

In [165]: y
Out[165]: [40.1, 40.2, 40.1, 40.2, 40.3]

In [166]: frame = [5,6,8,11,13]

In [167]: extend_arrs(x,y,frame)
Out[167]: 
(array([ 5,  6,  7,  8,  9, 10, 11, 12, 13]),
 array([80.1, 80.2, 80.2, 80.1, 80.1, 80.1, 80.2, 80.2, 80.3]),
 array([40.1, 40.2, 40.2, 40.1, 40.1, 40.1, 40.2, 40.2, 40.3]))

# Output in tabular format for quick reference
In [168]: np.c_[extend_arrs(x,y,frame)]
Out[168]: 
array([[ 5. , 80.1, 40.1],
       [ 6. , 80.2, 40.2],
       [ 7. , 80.2, 40.2],
       [ 8. , 80.1, 40.1],
       [ 9. , 80.1, 40.1],
       [10. , 80.1, 40.1],
       [11. , 80.2, 40.2],
       [12. , 80.2, 40.2],
       [13. , 80.3, 40.3]])

Answer 3

Numpy和diff

使用diff来判断我们需要重复多少次。

函数

def ext_pir(x, y, frame):
  x, y, frame = map(np.asarray, [x, y, frame])

  mn, mx = frame.min(), frame.max() + 1
  d = np.diff(np.append(frame, mx))
  r = np.arange(len(frame))
  i = r.repeat(d)

  return x[i], y[i], np.arange(mn, mx)

游行示威

x = [80.1, 80.2, 80.1, 80.2, 80.3]
y = [40.1, 40.2, 40.1, 40.2, 40.3]

frame = [1, 2, 3, 4, 6]

print("Inputs")
print(*map(np.asarray, (x, y, frame)), sep='\n')
print()
print("Outputs")
print(*ext_pir(x, y, frame), sep='\n')

Inputs
[80.1 80.2 80.1 80.2 80.3]
[40.1 40.2 40.1 40.2 40.3]
[1 2 3 4 6]

Outputs
[80.1 80.2 80.1 80.2 80.2 80.3]
[40.1 40.2 40.1 40.2 40.2 40.3]
[1 2 3 4 5 6]

x = [80.1, 80.2, 80.1, 80.2, 80.3]
y = [40.1, 40.2, 40.1, 40.2, 40.3]

frame = [5, 6, 8, 11, 13]

print("Inputs")
print(*map(np.asarray, (x, y, frame)), sep='\n')
print()
print("Outputs")
print(*ext_pir(x, y, frame), sep='\n')

Inputs
[80.1 80.2 80.1 80.2 80.3]
[40.1 40.2 40.1 40.2 40.3]
[ 5  6  8 11 13]

Outputs
[80.1 80.2 80.2 80.1 80.1 80.1 80.2 80.2 80.3]
[40.1 40.2 40.2 40.1 40.1 40.1 40.2 40.2 40.3]
[ 5  6  7  8  9 10 11 12 13]