import psycopg2
import time
def read_database():
conn = None
try:
conn = psycopg2.connect(database="capitadb", user="capita_user", password="capita_user",
host = "127.0.0.1", port = "5432")
cur = conn.cursor()
start_time = time.time()
cur.execute("COPY stagging(Activity_ID,F_Qtr,Fiscal_Week_Num,Manager,MBadge) FROM '/home/vivek/Downloads/dell_data.csv' DELIMITER',' CSV;;")
print("--- %s seconds ---" % (time.time() - start_time))
print("Operation done successfully")
conn.commit()
except Exception as e:
print("Error: %s" % e)
finally:
conn.close()
if __name__ == '__main__':
read_database()
这里我们在csv文件中有15列,但是我们只想复制4列。我们如何在不提取任何文件中的数据的情况下实现这一点?你知道吗
您需要使用
COPY FROM STDIN
功能-http://initd.org/psycopg/docs/cursor.html#cursor.copy_from。您将能够为该函数提供类似文件的对象。您可以使用itertools模块来实现这一点相关问题 更多 >
编程相关推荐