如何查看Kafka中的分区数量(Confluent_kafka)

from confluent_kafka.admin import AdminClient, NewTopic

bootstrap_servers = 'kafka服务器地址:端口号'
admin_client = AdminClient({'bootstrap.servers': bootstrap_servers})

def get_partition_count(topic_name):
    topic_metadata = admin_client.list_topics(topic_name)
    partitions = topic_metadata.topics[topic_name].partitions
    return len(partitions)

topic_name = '要查询的主题名称'
partition_count = get_partition_count(topic_name)
print("分区数量：", partition_count)