如何在列中选择前 3 个不同的值

Question

我正在尝试选择列中的前 3 个不同点，不包括第一个 0 点。我需要不同的点来执行多项式回归。

我的列：

  A = np.array([0, 0.5, 0, 0 ,1.0, 2.0])

预期：

  B = np.array([0, 0.5, 0.0, 1.0])

基本上我想跳过一个 0 而选择 1.0。

Answer 1

import numpy as np

A = np.array([0, 0.5, 0, 0 ,1.0, 2.0])
B = A[0]
B = np.append(B, list(set(A[1:]))[:3])
print(B)