python 3，pandas数据帧

Question

我有一个数据框，我需要创建一个新列并分配值。如果说该列中有100行，我希望能够说前50个被赋值为1，其余为0.我知道为了创建新列，我可以做df ['CLASS' ] = 0，但在这种情况下，所有值都是0。我应该如何修改此语句以便能够分配不同的值？

Answer 1

您可以在通过字典创建数据框时指定值，如下所示：

df = pd.DataFrame({'CLASS': [1]*50 + [0]*50})

Answer 2

你可以做：

df = pd.DataFrame(np.random.rand(100,3))
df['CLASS']=[1]*50 + [0]*50