」工欲善其事,必先利其器。「—孔子《論語.錄靈公》
首頁 > 程式設計 > 將大量記錄插入 MySQL 表的 Python 程式碼。

將大量記錄插入 MySQL 表的 Python 程式碼。

發佈於2024-08-28
瀏覽:817

Python code that inserts a large number of records into a MySQL table.

此 Python 程式碼會建立一個名為 my_table 的 MySQL 表,其中包含名稱、年齡和城市列。

然後,它會在表中插入 100 萬筆記錄以及隨機資料以用於演示目的。

import mysql.connector
import random

# Database configuration
db_config = {
    'host': '127.0.0.1',
    'port': 3309,
    'user': 'my_user',
    'password': 'my_password',
    'database': 'my_database'
}

# Function to create connection and insert records
def insert_records(num_records):
    try:
        connection = mysql.connector.connect(**db_config)
        cursor = connection.cursor()

        for i in range(num_records):
            # Generate random data for demonstration
            name = f'Name{i}'
            age = random.randint(18, 80)
            city = f'City{i % 100}'  # Only 100 cities for simplicity

            # Insert record into the table
            cursor.execute("INSERT INTO my_table (name, age, city) VALUES (%s, %s, %s)", (name, age, city))

        connection.commit()
        print(f"{num_records} records inserted successfully")
    except mysql.connector.Error as error:
        print("Error inserting records:", error)
    finally:
        if connection.is_connected():
            cursor.close()
            connection.close()

# Number of records to insert
num_records = 1000000  # Inserting 1 million records

# Create table if not exists
create_table_query = '''
CREATE TABLE IF NOT EXISTS my_table (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(255),
    age INT,
    city VARCHAR(255)
)
'''

try:
    connection = mysql.connector.connect(**db_config)
    cursor = connection.cursor()
    cursor.execute(create_table_query)
    print("Table 'my_table' created successfully")
except mysql.connector.Error as error:
    print("Error creating table:", error)
finally:
    if connection.is_connected():
        cursor.close()
        connection.close()

# Insert records
insert_records(num_records)

在運行此程式碼之前,請確保定義
‘主機’: ‘127.0.0.1’,
‘端口’:3309,
‘用戶’:‘我的用戶’,
‘密碼’:‘我的密碼’,
‘資料庫’:‘我的資料庫’
與您的實際 MySQL 憑證和資料庫名稱。

另外,請確保安裝了 mysql-connector-python 軟體包

pip 安裝 mysql-connector-python

dmi@dmi-laptop:~/my_mysql_postgres$ pip install mysql-connector-python
Defaulting to user installation because normal site-packages is not writeable
Collecting mysql-connector-python
  Downloading mysql_connector_python-8.4.0-cp310-cp310-manylinux_2_17_x86_64.whl (19.4 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 19.4/19.4 MB 3.8 MB/s eta 0:00:00
Installing collected packages: mysql-connector-python
Successfully installed mysql-connector-python-8.4.0

例子:

mysql> select count(1) from my_table;
 ---------- 
| count(1) |
 ---------- 
|  1000000 |
 ---------- 
1 row in set (0.05 sec)

mysql> 

mysql> select * from my_table limit 10;
 ---- ------- ------ ------- 
| id | name  | age  | city  |
 ---- ------- ------ ------- 
|  1 | Name0 |   38 | City0 |
|  2 | Name1 |   49 | City1 |
|  3 | Name2 |   27 | City2 |
|  4 | Name3 |   64 | City3 |
|  5 | Name4 |   19 | City4 |
|  6 | Name5 |   63 | City5 |
|  7 | Name6 |   36 | City6 |
|  8 | Name7 |   42 | City7 |
|  9 | Name8 |   51 | City8 |
| 10 | Name9 |   54 | City9 |
 ---- ------- ------ ------- 
10 rows in set (0.01 sec)

mysql> 

[email protected]

版本聲明 本文轉載於:https://dev.to/dm8ry/python-code-that-inserts-a-large-number-of-records-into-a-mysql-table-124c?1如有侵犯,請聯絡study_golang @163.com刪除
最新教學 更多>

免責聲明: 提供的所有資源部分來自互聯網,如果有侵犯您的版權或其他權益,請說明詳細緣由並提供版權或權益證明然後發到郵箱:[email protected] 我們會在第一時間內為您處理。

Copyright© 2022 湘ICP备2022001581号-3