如何在列中选择前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)