将最后一个值替换为之前的值

我有一组不完整的三元组：

a = np.arange(10) // 3 + 1
print (a)
[1 1 1 2 2 2 3 3 3 4]

Or:

a1 = np.arange(11) // 3 + 1
print (a1)
[1 1 1 2 2 2 3 3 3 4 4]

最后的值为4，但我需要将其更改为之前的，这里3，因为不是三元组。

I try:

unique, counts = np.unique(a, return_counts=True)
print (unique)
[1 2 3 4]

print (counts)
[3 3 3 1]

预期输出a:

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

and for a1:

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

如果全部都是三元组，则不更改任何值：

a2 = np.arange(12) // 3 + 1
print (a2)

[1 1 1 2 2 2 3 3 3 4 4 4]

我需要一些通用的解决方案，因为三元组应该改为4或者更多。

这是一种方法 -

def replace_irregular_groupings(a, N=3):
    n = len(a)
    m = N*(n//N)
    if m!=n:
        a[m:] = a[m-1]
    return a

样本运行 -

In [125]: replace_irregular_groupings(np.arange(9) // 3 + 1)
Out[125]: array([1, 1, 1, 2, 2, 2, 3, 3, 3])

In [126]: replace_irregular_groupings(np.arange(10) // 3 + 1)
Out[126]: array([1, 1, 1, 2, 2, 2, 3, 3, 3, 3])

In [127]: replace_irregular_groupings(np.arange(11) // 3 + 1)
Out[127]: array([1, 1, 1, 2, 2, 2, 3, 3, 3, 3, 3])

In [128]: replace_irregular_groupings(np.arange(12) // 3 + 1)
Out[128]: array([1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4])