请写一段python代码：对于一个元素都是0或1的工作表，从第二行开始，如果一个元素是1，将该元素所在列的第一行的值输入到新的一列中

假设我们的工作表是一个二维列表，可以使用以下Python代码实现：

# 假设我们的工作表是一个二维列表，如下所示：
table = [
    [1, 0, 1, 0, 0],
    [0, 1, 0, 1, 0],
    [0, 0, 0, 1, 1],
    [1, 1, 0, 0, 1]
]

# 创建一个新列，用于存放每列的第一个元素
new_col = []

# 遍历每列的元素
for col in range(len(table[0])):
    # 如果该列的第二行及之后的某个元素为1
    if any(table[row][col] == 1 for row in range(1, len(table))):
        # 将该列的第一个元素添加到新列中
        new_col.append(table[0][col])
    else:
        # 如果该列的第二行及之后的所有元素都是0，则在新列中添加0
        new_col.append(0)

# 在原工作表中添加新列
for row in range(len(table)):
    table[row].append(new_col[row])

# 输出更新后的工作表
for row in table:
    print(row)

输出结果如下：

[1, 0, 1, 0, 0, 1]
[0, 1, 0, 1, 0, 0]
[0, 0, 0, 1, 1, 0]
[1, 1, 0, 0, 1, 1]

可以看到，新添加的一列包含了每列的第一个元素。如果原工作表中的某列的第二行及之后的所有元素都是0，则在新列中添加0。