将C++ CRC16 modbus rtu计算器移植到python

#include <iostream>
#include <vector>

unsigned short crc16_modbus(const std::vector<unsigned char>& data) {
    unsigned short crc = 0xFFFF;
    for (unsigned char byte : data) {
        crc ^= byte;
        for (int i = 0; i < 8; ++i) {
            if (crc & 0x0001) {
                crc >>= 1;
                crc ^= 0xA001;
            } else {
                crc >>= 1;
            }
        }
    }
    return crc;
}

int main() {
    std::vector<unsigned char> data = {0x01, 0x03, 0x00, 0x00, 0x00, 0x0A};
    unsigned short crc = crc16_modbus(data);
    std::cout << "CRC: " << std::hex << crc << std::endl;
    return 0;
}

def crc16_modbus(data):
    crc = 0xFFFF
    for byte in data:
        crc ^= byte
        for _ in range(8):
            if crc & 0x0001:
                crc >>= 1
                crc ^= 0xA001
            else:
                crc >>= 1
    return crc

if __name__ == "__main__":
    data = [0x01, 0x03, 0x00, 0x00, 0x00, 0x0A]
    crc = crc16_modbus(data)
    print(f"CRC: {crc:04X}")