在csv文件中写入从tweet中提取的url

import csv
import re

# 从tweet中提取url的函数
def extract_urls_from_tweet(tweet):
    # 使用正则表达式来匹配url
    urls = re.findall(r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\\(\\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', tweet)
    return urls

# 假设有一个tweets列表存储了多条tweet文本
tweets = [
    "Check out this awesome website: http://example.com",
    "Here's a cool link: https://www.google.com",
    "Don't forget to visit our site: http://example.org"
]

# 创建csv文件并写入提取到的url
with open('tweet_urls.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    
    # 写入表头
    writer.writerow(['URL'])
    
    # 遍历tweets列表并提取url，将url写入csv文件中
    for tweet in tweets:
        urls = extract_urls_from_tweet(tweet)
        for url in urls:
            writer.writerow([url])

# 输出操作完成提示
print("URLs have been written to tweet_urls.csv file.")