def main():

    import happybase
    pool = happybase.ConnectionPool(size=10, host='hadoop-master', port=9090)
    with pool.connection() as conn:
        article_similar = conn.table('article_similar')
        data = article_similar.row(str(116644).encode(), columns=[b'similar'])
        print(data)
        s_d = sorted(data.items(), key=lambda item: item[1])
        # 取出前K个相似
        reco = [int(i[0].split(b':')[1]) for i in s_d][:10]
        print(reco)
        conn.close()


if __name__ == '__main__':
    main()