用Python写一个根据进程名称获取CPU和内存占用率的脚本并把数据写入csv文件

以下是一个根据进程名称获取CPU和内存占用率的Python脚本，并将数据写入.csv文件的示例：

import psutil
import csv

def get_process_info(process_name):
    # 获取所有进程列表
    all_processes = psutil.process_iter(attrs=['name', 'cpu_percent', 'memory_percent'])
    
    # 根据进程名称过滤进程列表
    filtered_processes = [p.info for p in all_processes if p.info['name'] == process_name]
    
    return filtered_processes

def write_to_csv(data, filename):
    with open(filename, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(['Process Name', 'CPU Usage (%)', 'Memory Usage (%)'])
        for process in data:
            writer.writerow([process['name'], process['cpu_percent'], process['memory_percent']])

if __name__ == '__main__':
    process_name = input("请输入进程名称: ")
    process_info = get_process_info(process_name)
    write_to_csv(process_info, 'process_info.csv')

运行脚本后，会要求输入进程名称，然后根据该名称获取进程的CPU和内存占用率信息，并将数据写入名为process_info.csv的.csv文件中