温馨提示×

Python怎么删除指定元素的行

小亿
87
2023-12-21 00:02:48
栏目: 编程语言

要删除指定元素的行,你可以使用列表解析来过滤出不包含该元素的行。以下是一个示例代码:

def remove_rows(data, element):
    return [row for row in data if element not in row]

data = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

element = 4

data = remove_rows(data, element)
print(data)

输出:

[[1, 2, 3], [7, 8, 9]]

在这个例子中,remove_rows 函数接受一个二维列表 data 和一个指定的元素 element。它使用列表解析过滤掉包含指定元素的行,然后返回过滤后的结果。在主程序中,调用 remove_rows 函数并传入数据和要删除的元素,然后打印结果。

0