sql - SQL Server : Output data frame into a table 中的 R

Question

这可能有一个简单的答案，但我无法弄明白，因为我仍然对在 SQL Server 中使用 R 有所了解。我有一段代码从 SQL Server 表中读取数据，在 R 中执行并返回一个数据框。

execute sp_execute_external_script
    @language=N'R',
    @script=N'inp_dat=InputDataSet
    inp_dat$NewCol=max(inp_dat$col1,inp_dat$col2)
    new_dat=inp_dat
    OutputDataSet=new_dat'
    @input_data_1=N'select * from IM_COMP_TEST_SQL2016.dbo.temp_table';

我想将 new_dat 插入到 SQL Server 表中(select * into new_table from new_dat)。我该怎么做？

Answer 1

如图所示tutorial ，您可以在先前创建的表中使用 INSERT INTO ... EXEC，其中的列与脚本的数据帧返回对齐:

INSERT INTO Table1
execute sp_execute_external_script
    @language=N'R',
    @script=N'inp_dat <- InputDataSet
              inp_dat$NewCol <- max(inp_dat$col1,inp_dat$col2)
              new_dat <- inp_dat',
    @input_data_1=N'SELECT * FROM IM_COMP_TEST_SQL2016.dbo.temp_table',
    @output_data_1=N'newdat';

但是，要使用生成表查询，可能需要 OPENQUERY() 或 OPENROWSET() 使用此 SO Post 中描述的临时分布式查询返回存储过程的输出:

存储过程

CREATE PROCEDURE dbo.R_DataFrame

AS

BEGIN
    execute sp_execute_external_script
        @language=N'R',
        @script=N'inp_dat <- InputDataSet
                  inp_dat$NewCol <- max(inp_dat$col1,inp_dat$col2)
                  new_dat <- inp_dat',
        @input_data_1=N'SELECT * FROM IM_COMP_TEST_SQL2016.dbo.temp_table',
        @output_data_1=N'newdat';

        -- ADD ALL COLUMN TYPES;
        WITH RESULT SETS (("newdat" [col1] varchar(20), [col2] double, [col3] int ...));
END
GO

Action 查询

SELECT * INTO Table1 
FROM OPENROWSET('SQLNCLI', 'Server=(local);Trusted_Connection=yes;',
                'EXEC dbo.R_DataFrame')