php和python获取股票代码前缀

<?php

function getExchange($code) {
    $prefix = substr($code, 0, 3);
    if ($prefix == '600' || $prefix == '601' || $prefix == '603') {
        return 'sh' . $code;
    } elseif ($prefix == '900') {
        return 'sh' . $code;
    } elseif ($prefix == '000' || $prefix == '001' || $prefix == '002') {
        return 'sz' . $code;
    } elseif ($prefix == '200') {
        return 'sz' . $code;
    } elseif ($prefix == '730') {
        return 'sz' . $code;
    } elseif ($prefix == '300') {
        return 'sz' . $code;
    } elseif ($prefix == '688') {
        return 'sh' . $code;
    } else {
        return '未知' . $code;
    }
}
echo getExchange('000858');  // 输出：深圳证券交易所

def getExchange(code):
    prefix = code[:3]
    if prefix in ('600', '601', '603', '900'):
        return 'sh' + code
    elif prefix in ('000', '001', '002', '200'):
        return 'sz' + code
    elif prefix == '730':
        return 'sh' + code
    elif prefix == '300':
        return 'sz' + code
    elif prefix == '688':
        return 'sh' + code
    else:
        return '未知' + code

print(getExchange('000858');)