python 3，熊猫数据框

Question

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

Answer 1

你可以做：

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

Answer 2

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

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