Как загрузить информацию в два столбца?

Question

Вопрос 1 : в файле phone.txt хранятся строки в формате код: номер

import pandas as pd
import sqlite3

con = sqlite3.connect('database.db')
data = pd.read_csv('phone.txt', sep='\t', header=None)
data.to_sql('post_table', con, if_exists='replace', index=False)

Я хочу загрузить все данные из файла phone.txt в базу данных database.db .Но у меня все загружено в одну колонку.А мне нужно загрузить в два столбца:

1. код
2. число

Как это сделать?

Вопрос2 : после загрузки информации в базу данных, как найти число по коду ?Например, если я хочу узнать, что число код = 7 (ответ: 9062621390).

Answer 1

Вопрос 1

В вашем примере pandas не может различить code и number, поскольку ваш файл разделен :.При чтении вашего файла вам нужно изменить разделитель на :, а также указать столбцы, поскольку ваш csv, похоже, не имеет такого заголовка

data = pd.read_csv('phone.txt',
                   sep=':',
                   names=['code', 'number'])

Вопрос 2

После помещения ваших данных в базу данных вы можете запросить их следующим образом

number = pd.read_sql_query('SELECT number FROM post_table WHERE code = (?)',
                           con,
                           params=(code,))

, где con - ваше sqlite-соединение.