在python中使用sqlite3计算跨列的非重复值

6ss1mwsb 于 2023-03-13 发布在 SQLite

关注(0)|答案(4)|浏览(186)

我尝试在python中使用sqlite3计算不同列的非重复值，但似乎无法得到正确的结果。我只能得到1列中非重复值的计数。
我创建了一个数据库，并将csv文件作为表导入

conn = sqlite3.connect('test.db')
curr = conn.cursor()

curr.execute('DROP TABLE IF EXISTS test')
curr.execute('CREATE TABLE social_table (from_id INTEGER, to_id INTEGER)')
conn.commit()

with open ('test.csv') as f:
    reader = csv.reader(f)
    next(reader, None)

    for row in reader:
        from_id = row[0]
        to_id = row[1]

        curr.execute('INSERT INTO test (from_id, to_id) VALUES (?, ?)',
                    (from_id, to_id))
    conn.commit()

| 发件人ID|目标标识符|
| - ------|- ------|
| 无|1个|
| 无|第二章|
| 无|三个|
| 无|四个|
| 无|五个|
| 无|六个|
| 无|七|
| 无|八个|
| 无|十个|
| 无|十一|
此表中跨列的唯一用户应为11，但我的代码

curr.execute("Select Count(*) from (Select DISTINCT from_id, to_id from test)")

给了我10块。
有人能帮我吗？

sqlite

来源：https://stackoverflow.com/questions/75704989/counting-distinct-values-across-columns-using-sqlite3-in-python

4条答案

按热度按时间

vsaztqbk1#

似乎您的查询不正确，请尝试以下操作：

curr.execute("""
    SELECT COUNT(DISTINCT user_id) 
    FROM (
        SELECT from_id as user_id FROM social_table 
        UNION 
        SELECT to_id as user_id FROM social_table
    )
""")

只需选择from_id的所有非重复值，然后将它们与to_id的所有非重复值进行并集，然后计算并集中非重复值的数量，这样就可以给予两列中唯一用户的总数。

赞(0）回复(0）举报 2023-03-13

fxnxkyjh2#

您应该同时计算from_id和to_id列中的非重复用户。我们可以使用联合来实现此目的：

SELECT COUNT(DISTINCT id) AS cnt
FROM
(
    SELECT from_id AS id FROM social_table
    UNION ALL
    SELECT to_id FROM social_table
) t;

赞(0）回复(0）举报 2023-03-13

0tdrvxhp3#

另一个想法是将列连接为字符串

from_id || '-' || to_id

因此，您的查询将是

curr.execute("Select Count(*) from (Select DISTINCT from_id || '-' || to_id  from social_table)")

赞(0）回复(0）举报 2023-03-13

iyr7buue4#

我认为from_id和to_id不应该为空，因此应该将表定义更正为：

CREATE TABLE social_table (from_id INTEGER NOT NULL, to_id INTEGER NOT NULL);

在这种情况下，您应该使用简单的COUNT(*)，而不是效率较低的COUNT(DISTINCT ...)：

sql = """
    SELECT COUNT(*) AS count 
    FROM (
      SELECT from_id FROM social_table 
      UNION 
      SELECT to_id FROM social_table
    )
"""
curr.execute(sql)

因为UNION从结果集中删除了重复项。
请参见demo。

赞(0）回复(0）举报 2023-03-13

我来回答

在python中使用sqlite3计算跨列的非重复值

4条答案

相关问题

热门标签

最新问答