时间戳

首页 > 解决方案 > 如何在 Databrick 中截断和/或使用通配符

pyspark - 如何在 Databrick 中截断和/或使用通配符

问题描述

我正在尝试在 databricks 中编写一个脚本，该脚本将根据文件名中的某些字符或仅根据文件中的日期戳来选择文件。

例如，以下文件如下所示：

LCMS_MRD_Delta_LoyaltyAccount_1992_2018-12-22 06-07-31

我在 Databricks 中创建了以下代码：

import datetime
now1 = datetime.datetime.now()
now = now1.strftime("%Y-%m-%d")

使用上面的代码，我尝试使用以下方法选择文件：

LCMS_MRD_Delta_LoyaltyAccount_1992_%s.csv'% now

但是，如果您仔细观察，您会注意到日期戳和时间戳之间有一个空格，即在 22 和 06 之间

LCMS_MRD_Delta_LoyaltyAccount_1992_2018-12-22 06 -07-31

这是因为如果这个空间阻止了我上面的代码工作。

我认为 Databricks 不支持通配符，因此以下内容不起作用：

LCMS_MRD_Delta_LoyaltyAccount_1992_%s.csv'% now

有人曾经建议截断时间戳。

有人可以让我知道是否：

A.TRUNCATING 将解决这个问题 B.我的代码有办法吗LCMS_MRD_Delta_LoyaltyAccount_1992_%s.csv'% now

选择整个文件？请记住，我绝对需要根据当前日期进行选择。我只想能够使用我的代码在文件上进行选择。

标签： pysparkpyspark-sqldatabricksazure-databricks

解决方案

您可以使用 dbutils 读取文件名，并可以在 if 语句中检查模式是否匹配：如果现在在 filname 中。因此，您无需直接读取具有特定模式的文件，而是获取文件列表，然后复制与所需模式匹配的具体文件。

以下代码适用于 databricks python 笔记本：

1. 将三个文件写入文件系统：

data = """
{"a":1, "b":2, "c":3}
{"a":{, b:3} 
{"a":5, "b":6, "c":7}

"""

dbutils.fs.put("/mnt/adls2/demo/files/file1-2018-12-22 06-07-31.json", data, True)
dbutils.fs.put("/mnt/adls2/demo/files/file2-2018-02-03 06-07-31.json", data, True)
dbutils.fs.put("/mnt/adls2/demo/files/file3-2019-01-03 06-07-31.json", data, True)

2. 将电影名读取为列表：

文件 = dbutils.fs.ls("/mnt/adls2/demo/files/")

3.获取实际日期：

import datetime

now = datetime.datetime.now().strftime("%Y-%m-%d")
print(now)

输出：2019-01-03

4.复制实际文件：

for i in range (0, len(files)):
  file = files[i].name
  if now in file:  
    dbutils.fs.cp(files[i].path,'/mnt/adls2/demo/target/' + file)
    print ('copied     ' + file)
  else:
    print ('not copied ' + file)

输出：

未复制 file1-2018-12-22 06-07-31.json

未复制 file2-2018-02-03 06-07-31.json

复制文件 3-2019-01-03 06-07-31.json

推荐阅读