如何将CSV数据集导入Prolog并作为知识库使用

我需要把一个100行、43列（包含20个可自定义名称属性）的CSV文件加载到Prolog中作为知识库，但不清楚具体操作方法。网上查阅资料后，未找到匹配该复杂需求的方案。以下是我编写的用于查看CSV行数据的代码：

:- use_module(library(csv)).

% Read the data from the CSV file
read_csv_file(File, Rows) :- 
    csv_read_file(File, Rows, [skip_header_row(true)]).

% Entry point for the program
main :- 
    % Open the CSV file
    open('data.csv', read, Stream),

    % Read the data from the file
    read_csv_file(Stream, Rows),

    % Close the file
    close(Stream),

    % Print the rows
    print_rows(Rows).

% Print the rows of the CSV file
print_rows([]).
print_rows([Row | Rest]) :- 
    writeln(Row),
    print_rows(Rest).

代码优化与自定义属性处理 #

1. 简化文件操作

原代码手动调用open/close属于冗余操作，csv_read_file/3可以直接接收文件名，内部会自动管理文件流，简化后的基础读取代码如下：

:- use_module(library(csv)).

read_csv_file(File, Rows) :-
    csv_read_file(File, Rows, [skip_header_row(true)]).

print_rows([]).
print_rows([Row | Rest]) :-
    writeln(Row),
    print_rows(Rest).

main :-
    read_csv_file('data.csv', Rows),
    print_rows(Rows).

2. 自定义属性映射与知识库构建

针对你需要自定义20个属性名的需求，可以通过header/1选项指定自定义表头，将CSV行转换为结构化的row/43项（匹配43列的结构），并将这些数据断言为Prolog事实，构建可查询的知识库：

:- use_module(library(csv)).

% 读取CSV并使用自定义表头转换数据
read_csv_with_custom_headers(File, CustomHeaders, Rows) :-
    csv_read_file(File, Rows, [
        skip_header_row(true),
        header(CustomHeaders)  % 传入自定义属性列表
    ]).

% 将行数据断言为知识库事实
assert_rows([]).
assert_rows([Row | Rest]) :-
    assertz(Row),
    assert_rows(Rest).

main :-
    % 定义自定义属性列表（前20个为你需要的自定义名称，剩余列可自行命名）
    CustomHeaders = [user_id, username, email, phone, address, 
                     attr6, attr7, attr8, attr9, attr10,
                     attr11, attr12, attr13, attr14, attr15,
                     attr16, attr17, attr18, attr19, attr20,
                     col21, col22, col23, col24, col25,
                     col26, col27, col28, col29, col30,
                     col31, col32, col33, col34, col35,
                     col36, col37, col38, col39, col40,
                     col41, col42, col43],
    read_csv_with_custom_headers('data.csv', CustomHeaders, Rows),
    assert_rows(Rows),
    writeln("CSV数据已成功加载为知识库事实").

3. 知识库查询示例

加载完成后，你可以直接通过结构化的事实进行查询，比如查找特定用户ID的数据：

% 查询user_id为1001的用户信息
?- row(1001, Username, Email, _, _, ...).

内容的提问来源于stack exchange，提问作者candy