请写一段python代码:对于一个元素都是0或1的工作表,将每一个是1的变量的索引名输入到新的一列中,
时间: 2024-06-11 09:06:57 浏览: 46
Python创建一个元素都为0的列表实例
假设工作表的数据是二维列表,可以使用以下代码实现:
```python
# 假设数据存储在 data 变量中,每行的元素为0或1
# 先创建一个空列表,用于存储每一行中值为1的元素的索引
new_column = []
# 遍历每一行
for row in data:
# 创建一个空列表,用于存储当前行中值为1的元素的索引
idx_list = []
# 遍历当前行中的每个元素
for i, val in enumerate(row):
# 如果元素值为1,将其索引添加到 idx_list 中
if val == 1:
idx_list.append(str(i))
# 将当前行中值为1的元素的索引拼接成一个字符串,添加到 new_column 中
new_column.append(','.join(idx_list))
# 将 new_column 添加到数据中作为新的一列
for i in range(len(data)):
data[i].append(new_column[i])
```
以上代码可以将每行中值为1的元素的索引,以逗号分隔的形式,添加到数据的最后一列。
阅读全文