如何使用pyqt4删除表格小部件中的特定行框

from PyQt4 import QtGui, QtCore

table_widget = QtGui.QTableWidget()
table_widget.setRowCount(row_count)
table_widget.setColumnCount(column_count)

for row in range(row_count):
    for column in range(column_count):
        item = QtGui.QTableWidgetItem()
        item.setText(data[row][column])
        table_widget.setItem(row, column, item)

button = QtGui.QPushButton("删除行")
button.clicked.connect(delete_row)

def delete_row():
    selected_rows = table_widget.selectionModel().selectedRows()
    for row in selected_rows:
        table_widget.removeRow(row.row())

layout = QtGui.QVBoxLayout()
layout.addWidget(table_widget)
layout.addWidget(button)

main_window = QtGui.QWidget()
main_window.setLayout(layout)
main_window.show()