【发布时间】:2019-07-23 21:35:49
【问题描述】:
我有一个 sqlite3 表,其中包含 0 和 1。 我想获取值为 1 的列名。 我已经尝试了 description 和 pragma 方法,但由于某种原因结果不正确。
以下是我的尝试:
.description
allResults = c.execute("SELECT * FROM db_1.hope5 WHERE Vehicle = '{}'".format(veh))
all_columnNames = [tup[0] for tup in allResults.description]
inputs_columnNames = all_columnNames[9:]
inputcheck = [tup[9:] for tup in allResults.fetchall()]
inputSelected = [val for i, val in enumerate(inputs_columnNames) if inputcheck[0][i] == 1]
编译指示
c = connection.cursor()
c.execute("PRAGMA TABLE_INFO(hope5)")
names = [tup[1] for tup in c.fetchall()]
names = names[2:]
result = c.execute("SELECT * FROM hope5 WHERE Vehicle = '{}' ".format(veh))
result = result.fetchall()
test = [names[index] for index, val in enumerate(result[0][2:]) if val==1]
桌子的形状
+----+-----+-----+-----+-----+-----+
| A | B | C | D | E | F |
+----+-----+-----+-----+-----+-----+
| 0 | 0 | 1 | 0 | 0 | 1 |
|----|-----|-----|-----|-----|-----|
| 1 | 0 | 0 | 0 | 1 | 0 |
+----+-----+-----+-----+-----+-----+
根据我上面的代码,结果需要是:
test = [A,C,E,F]
【问题讨论】:
-
@MichaelBianconi 链接是 SQL server 而不是 python & Sqlite3!